nir/spirv: Use the type from the deref for atomics