From f47342f802661eb8b5d1de9b9fb7667b127bd4b2 Mon Sep 17 00:00:00 2001
From: tocean <yuxiangyang@microsoft.com>
Date: Tue, 30 Jan 2024 07:00:28 +0000
Subject: [PATCH] build succeed on Nvidia GPU

---
 csrc/extensions/operators/arithmetic.cu | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/csrc/extensions/operators/arithmetic.cu b/csrc/extensions/operators/arithmetic.cu
index 3c97e5b7..cac7589b 100644
--- 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,