Physical uplink shared channel (PUSCH) transmission based on robust codebook

A base station, comprising:

• memory to store a codebook, wherein the codebook is predefined or configured based on different numbers of phase tracking reference signal (PT-RS) antenna ports (NPT-RS) and different waveforms and wherein the predefined or configured codebook is divided into a plurality of codebook subsets based on different numbers of PT-RS antenna ports and/or different waveforms;
• and
• processing circuitry, coupled with the memory, to:
• decode a UE capability of a user equipment (UE) reported by the UE, wherein the UE capability comprises a number of phase tracking reference signal (PT-RS) antenna ports (NPT-RS) supported by the UE, wherein the UE also stores the codebook;
• select a codebook subset within the codebook based on the NPT-RS and/or waveform configuration for the UE;
• transmit, to the UE, first control signaling comprising an indication of the codebook subset;
• select a precoder from the codebook subset, based on at least the NPT-RS;
• and
• transmit, to the UE, second control signaling to control physical uplink shared channel (PUSCH) transmission of the UE, wherein the second control signaling comprises at least one parameter to indicate the selected precoder.