Remove defer from hot path and ensure Atomic is applied consistently.
Remove defer from hot path and ensure Atomic is applied consistently.

PiperOrigin-RevId: 270114317

google/gvisor,google/gvisor,google/gvisor,google/gvisor,google/gvisor,google/gvisor,google/gvisor,google/gvisor