vrr/kvcache-ai-ktransformers
2
0
Fork 0
mirror of https://github.com/kvcache-ai/ktransformers.git synced 2025-09-09 13:55:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge pull request #904 from Azure-Tang/main

Browse source 
[fix]Fix rocm compilation
This commit is contained in:
Azure 2025-03-16 00:36:16 +08:00 • committed by GitHub
commit 63604cac59
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ktransformers/ktransformers_ext/cuda/custom_gguf/dequant.cu
View file

@ -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