Merge pull request #904 from Azure-Tang/main

[fix]Fix rocm compilation

Merge pull request #904 from Azure-Tang/main
[fix]Fix rocm compilation
63604cac · Azure · GitHub · c51818c3 · 4a312373 · 63604cac
Unverified Commit 63604cac authored Mar 16, 2025 by Azure Committed by GitHub Mar 16, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

ktransformers/ktransformers_ext/cuda/custom_gguf/dequant.cu ktransformers/ktransformers_ext/cuda/custom_gguf/dequant.cu +1 -1

No files found.
--- a/ktransformers/ktransformers_ext/cuda/custom_gguf/dequant.cu
+++ b/ktransformers/ktransformers_ext/cuda/custom_gguf/dequant.cu
@@ -16,7 +16,7 @@
 #include <cstdint>
 #include <c10/cuda/CUDAGuard.h>

-#ifdef KTRANSFORMERS_USE_ROCM
+#ifdef __HIP_PLATFORM_AMD__
 typedef hip_bfloat16 nv_bfloat16;
 #endif