SRS resource set and beam order association for multi-beam PUSCH

An apparatus for wireless communication at a network entity, comprising:

• memory;
• and
• at least one processor coupled to the memory and, based at least in part on stored information that is stored in the memory, the at least one processor is configured to cause the apparatus to:
• transmit a configuration of a first sounding reference signal (SRS) resource set and a second SRS resource set;
• transmit, for a user equipment (UE), downlink control information (DCI) for an uplink (UL) transmission, the DCI indicating the first SRS resource set and the second SRS resource set, wherein the DCI comprises a codepoint representing an order for the first SRS resource set and the second SRS resource set, wherein the codepoint is selected from a set of four codepoint values including:
• a first codepoint value that indicates the first SRS resource set is to be used,
• a second codepoint value that indicates the second SRS resource set is to be used,
• a third codepoint value that indicates that the first SRS resource set is to be used for a first set of repetitions, and the second SRS resource set is to be used for a second set of repetitions, and
• a fourth codepoint value that indicates that the second SRS resource set is to be used for the first set of repetitions, and the first SRS resource set is to be used for the second set of repetitions;
• and
• receive, based on the DCI indicating the third codepoint value or the fourth codepoint value, the first set of repetitions of a physical uplink shared channel (PUSCH) based on the first SRS resource set and the second set of repetitions of the PUSCH based on the second SRS resource set in the order indicated by the codepoint of the DCI.

A method of wireless communication at a user equipment (UE), comprising:

• receiving a configuration of a first sounding reference signal (SRS) resource set and a second SRS resource set;
• receiving, from a network entity, downlink control information (DCI) for an uplink (UL) transmission, the DCI indicating the first SRS resource set and the second SRS resource set, wherein the DCI comprises a codepoint representing an order for the first SRS resource set and the second SRS resource set, wherein the codepoint is selected from a set of four codepoint values including:
• a first codepoint value that indicates the first SRS resource set is to be used,
• a second codepoint value that indicates the second SRS resource set is to be used,
• a third codepoint value that indicates that the first SRS resource set is to be used for a first set of repetitions, and the second SRS resource set is to be used for a second set of repetitions, and
• a fourth codepoint value that indicates that the second SRS resource set is to be used for the first set of repetitions, and the first SRS resource set is to be used for the second set of repetitions;
• and
• transmitting, to the network entity based on the DCI indicating the third codepoint value or the fourth codepoint value, the first set of repetitions of a physical uplink shared channel (PUSCH) based on the first SRS resource set and the second set of repetitions of the PUSCH based on the second SRS resource set in the order indicated by the codepoint of the DCI.

A method for wireless communication at a network entity, comprising:

• transmitting a configuration of a first sounding reference signal (SRS) resource set and a second SRS resource set;
• transmitting, for a user equipment (UE), downlink control information (DCI) for an uplink (UL) transmission, the DCI indicating the first SRS resource set and the second SRS resource set, wherein the DCI comprises a codepoint representing an order for the first SRS resource set and the second SRS resource set, wherein the codepoint is selected from a set of four codepoint values including:
• a first codepoint value that indicates the first SRS resource set is to be used,
• a second codepoint value that indicates the second SRS resource set is to be used,
• a third codepoint value that indicates that the first SRS resource set is to be used for a first set of repetitions, and the second SRS resource set is to be used for a second set of repetitions, and
• a fourth codepoint value that indicates that the second SRS resource set is to be used for the first set of repetitions, and the first SRS resource set is to be used for the second set of repetitions;
• and
• receiving, based on the DCI indicating the third codepoint value or the fourth codepoint value, the first set of repetitions of a physical uplink shared channel (PUSCH) based on the first SRS resource set and the second set of repetitions of the PUSCH based on the second SRS resource set in the order indicated by the codepoint of the DCI.