fix rocm compilation

4a312373 · Azure-Tang · c51818c3 · 4a312373
Commit 4a312373 authored Mar 15, 2025 by Azure-Tang
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