Merge branch 'yuxiang/amd' of https://github.com/Azure/MS-AMP into yu…

…xiang/amd
Azure · Jan 30, 2024 · fcbabf9 · fcbabf9
2 parents 2bf6f89 + f47342f
commit fcbabf9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/csrc/extensions/operators/arithmetic.cu b/csrc/extensions/operators/arithmetic.cu
@@ -8,7 +8,9 @@
 
 namespace msamp {
 
+#ifdef __HIP_PLATFORM_AMD__
 __device__ msamp::DeviceSyncer device_syncer;
+#endif
 
 void add_to_fp8(at::Tensor fp8_tensor,
                 at::Tensor scale,