Various solutions with respect to codebook-based uplink transmission in wireless communications are described. A user equipment (UE) receives a signal indicating codebook subset restriction (CBSR) from a network node of a wireless network. The UE selects one or more codewords or a codebook from a master codebook based on the CBSR. The master codebook includes codewords from multiple components. The UE then performs a physical uplink shared channel (PUSCH) transmission to the network node using the one or more codewords or the codebook.