spirv/nir: Add support for OpAtomicLoad/Store