[AMD] Reorganize hip-related header files in sgl-kernel (#9320)

2025-08-18 16:53:44 -07:00
parent c2fbf60f39
commit c6c379ab31
14 changed files with 18 additions and 17 deletions
--- a/sgl-kernel/csrc/allreduce/mscclpp_allreduce.cuh
+++ b/sgl-kernel/csrc/allreduce/mscclpp_allreduce.cuh
@@ -1,7 +1,7 @@
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT license.
 #pragma once
-#if defined(__HIP_PLATFORM_AMD__)
+#ifdef USE_ROCM
 #include <hip/hip_fp16.h>
 #else
 #include <cuda_bf16.h>