Methods, systems, and devices for wireless communications are described. A first wireless device may transmit, to a second wireless device, capability control signaling indicating a first quantity of beam switches is supported within a first time period for a first antenna subarray, a first antenna module, or both, of the first wireless device. The first wireless device may receive, from the second wireless device in accordance with the capability control signaling, a grant scheduling one or more messages and indicating a set of beams of the first wireless device over which to communicate the one or more messages in respective transmission time intervals (TTIs) of the first time period. The first wireless device may then transmit or receive, over the first time period using the first antenna subarray, the first antenna module, or both, the one or more messages using the set of beams.