Various aspects of the present disclosure generally relate to wireless communication. In some aspects, a user equipment (UE) may receive configuration information associated with a first sounding reference signal (SRS) resource set, including a first quantity of SRS resources, and a second SRS resource set, including a second quantity of SRS resources. The UE may receive downlink control information (DCI) scheduling a spatial division multiplexing communication, the DCI indicating a first one or more SRS resources, associated with a first set of layers, and a second one or more SRS resources, associated with a second set of layers, from the first SRS resource set and/or the second SRS resource set based on: a dynamic switching indicator, a first SRS resource indicator (SRI), a second SRI, the first quantity of SRS resources, the second quantity of SRS resources, and/or a maximum rank. Numerous other aspects are described.