The present disclosure relates to methods and devices for wireless communication including an apparatus, e.g., a UE and/or a base station. The apparatus can receive SSB grouping information associated with one or more SSB groups, each of the one or more SSB groups including one or more TRPs. The apparatus can also receive at least one of TCI information or QCL information, each of the TCI information and the QCL information indicating an SSB of the one or more SSB groups that is associated with a downlink Tx beam for downlink data or an uplink Rx beam for uplink data. Additionally, the apparatus can communicate the downlink data from or the uplink data to TRPs in a same SSB group or TRPs in different SSB groups based on at least one of the SSB grouping information, the TCI information, or the QCL information.