Uplink rank adaptation for mimo communication

A method of wireless communication performed by a base station (BS), the method comprising:

• transmitting, to a user equipment (UE), a configuration indicating a first multi-port UL reference signal resource.
• receiving, from the UE, an uplink (UL) reference signal from one or more of a plurality of UL reference signal ports associated with a subset of the first multi-port UL reference signal resource;
• and
• transmitting, to the UE, an UL scheduling grant including a precoding information associated with a first quantity of transmission layers.

A user equipment (UE), comprising:

• a transceiver configured to:
• receive, from a base station (BS), a configuration indicating a first multi-port uplink (UL) reference signal resource associated with a plurality of UL reference signal ports;
• and
• a processor configured to:
• determine, based on at least one of a first overheating status at the UE, a first traffic throughput of the UE, or a first traffic latency of the UE, a first quantity of transmission layers for a first codebook-based transmission,
• wherein the transceiver is further configured to transmit, to the BS, an UL reference signal using one or more of the plurality of UL reference signal ports associated with the first multi-port UL reference signal resource based on the first quantity of transmission layers.