[WHLSL] Implement atomic operations and barriers