Add test for atomicrmw