conditionally include cuda_profiler_api.h (#219)

deepspeedai · Aug 16, 2023 · 58aed65 · 58aed65
1 parent c2b0736
commit 58aed65
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/megatron/fused_kernels/scaled_softmax_cuda.cu b/megatron/fused_kernels/scaled_softmax_cuda.cu
@@ -4,7 +4,9 @@
 #include <cuda.h>
 #include <cuda_runtime.h>
 #include <cuda_fp16.h>
+#ifndef __HIP_PLATFORM_HCC__
 #include <cuda_profiler_api.h>
+#endif
 #include <ATen/cuda/CUDAContext.h>
 #include <torch/extension.h>
 #include "scaled_masked_softmax.h"