A method for wireless communications by a user equipment, UE, (115, 605, 705, 905) comprising: receiving (1405, 1505) sounding reference signal resource set configuration information that indicates an aperiodic sounding reference signal resource set, an aperiodic downlink reference signal resource associated with the aperiodic sounding reference signal resource set, and a slot offset associated with the aperiodic downlink reference signal resource; receiving (1410, 1510) downlink control information in a first slot, the downlink control information indicating that the aperiodic sounding reference signal resource set is triggered; determining (1415, 1515) a quasi co-location assumption for the aperiodic downlink reference signal resource based at least in part on the slot offset associated with the aperiodic downlink reference signal resource; receiving and measuring (1420, 1525) the aperiodic downlink reference signal resource in a second slot indicated by the first slot and the slot offset based at least in part on the quasi co-location assumption; determining an uplink precoder for the aperiodic sounding reference signal resource set based on the measuring; and transmitting a sounding reference signal on the aperiodic sounding reference signal resource set based on the calculated uplink precoder, characterized in that the determining the quasi co-location assumption further comprises determining whether a beam switching capability of the UE is greater than a scheduling offset between the downlink control information and the aperiodic downlink reference signal resource, and when the beam switching capability of the UE is greater than the scheduling offset: detecting a downlink signal scheduled to overlap with the aperiodic downlink reference signal resource, wherein the quasi co-location assumption is associated with the downlink signal based at least in part on the beam switching capability being greater than the scheduling offset; or identifying a default quasi co-location assumption, wherein the quasi co-location assumption is the default quasi co-location assumption based at least in part on the beam switching capability being greater than the scheduling offset. receiving (1405, 1505) sounding reference signal resource set configuration information that indicates an aperiodic sounding reference signal resource set, an aperiodic downlink reference signal resource associated with the aperiodic sounding reference signal resource set, and a slot offset associated with the aperiodic downlink reference signal resource; receiving (1410, 1510) downlink control information in a first slot, the downlink control information indicating that the aperiodic sounding reference signal resource set is triggered; determining (1415, 1515) a quasi co-location assumption for the aperiodic downlink reference signal resource based at least in part on the slot offset associated with the aperiodic downlink reference signal resource; receiving and measuring (1420, 1525) the aperiodic downlink reference signal resource in a second slot indicated by the first slot and the slot offset based at least in part on the quasi co-location assumption; determining an uplink precoder for the aperiodic sounding reference signal resource set based on the measuring; and transmitting a sounding reference signal on the aperiodic sounding reference signal resource set based on the calculated uplink precoder, characterized in that the determining the quasi co-location assumption further comprises determining whether a beam switching capability of the UE is greater than a scheduling offset between the downlink control information and the aperiodic downlink reference signal resource, and when the beam switching capability of the UE is greater than the scheduling offset: detecting a downlink signal scheduled to overlap with the aperiodic downlink reference signal resource, wherein the quasi co-location assumption is associated with the downlink signal based at least in part on the beam switching capability being greater than the scheduling offset; or identifying a default quasi co-location assumption, wherein the quasi co-location assumption is the default quasi co-location assumption based at least in part on the beam switching capability being greater than the scheduling offset. detecting a downlink signal scheduled to overlap with the aperiodic downlink reference signal resource, wherein the quasi co-location assumption is associated with the downlink signal based at least in part on the beam switching capability being greater than the scheduling offset; or identifying a default quasi co-location assumption, wherein the quasi co-location assumption is the default quasi co-location assumption based at least in part on the beam switching capability being greater than the scheduling offset.