diff --git a/ktransformers/ktransformers_ext/cuda/custom_gguf/dequant.cu b/ktransformers/ktransformers_ext/cuda/custom_gguf/dequant.cu
index 2363035..3a6151b 100644
--- 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