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,