build succeed on Nvidia GPU

Azure · Jan 30, 2024 · 5d2f9b1 · 5d2f9b1
1 parent 6cc7af8
commit 5d2f9b1
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,