Skip to content

Commit 5d2f9b1

Browse files
toceanwenchenvincent
authored andcommitted
build succeed on Nvidia GPU
1 parent 6cc7af8 commit 5d2f9b1

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

csrc/extensions/operators/arithmetic.cu

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,9 @@
88

99
namespace msamp {
1010

11+
#ifdef __HIP_PLATFORM_AMD__
1112
__device__ msamp::DeviceSyncer device_syncer;
13+
#endif
1214

1315
void add_to_fp8(at::Tensor fp8_tensor,
1416
at::Tensor scale,

0 commit comments

Comments
 (0)