Merge pull request #904 from Azure-Tang/main

[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

View file

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