Skip to content

GitLab

Unverified Commit 63604cac authored by Azure's avatar Azure Committed by GitHub
Browse files

Merge pull request #904 from Azure-Tang/main 

[fix]Fix rocm compilation
parents c51818c3 4a312373
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
ktransformers/ktransformers_ext/cuda/custom_gguf/dequant.cu
View file @ 63604cac
...@@ -16,7 +16,7 @@ ...@@ -16,7 +16,7 @@
#include <cstdint> #include <cstdint>
#include <c10/cuda/CUDAGuard.h> #include <c10/cuda/CUDAGuard.h>
#ifdef KTRANSFORMERS_USE_ROCM #ifdef __HIP_PLATFORM_AMD__
typedef hip_bfloat16 nv_bfloat16; typedef hip_bfloat16 nv_bfloat16;
#endif #endif
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment