fix(hip): 添加hip_cooperative_groups.h头文件并注释__any_sync宏

This commit is contained in:
2026-01-16 15:42:14 +08:00
parent 83db9c468e
commit 9d7890f8c6

View File

@@ -5,7 +5,7 @@
#include <hipblas/hipblas.h> #include <hipblas/hipblas.h>
#include <hip/hip_fp16.h> #include <hip/hip_fp16.h>
#include <hip/hip_bf16.h> #include <hip/hip_bf16.h>
#include <hip/hip_cooperative_groups.h>
#if defined(GGML_HIP_ROCWMMA_FATTN) #if defined(GGML_HIP_ROCWMMA_FATTN)
#include <rocwmma/rocwmma-version.hpp> #include <rocwmma/rocwmma-version.hpp>
#endif // defined(GGML_HIP_ROCWMMA_FATTN) #endif // defined(GGML_HIP_ROCWMMA_FATTN)
@@ -32,7 +32,7 @@
#define __shfl_up_sync(mask, var, laneMask, width) __shfl_up(var, laneMask, width) #define __shfl_up_sync(mask, var, laneMask, width) __shfl_up(var, laneMask, width)
#define __shfl_xor_sync(mask, var, laneMask, width) __shfl_xor(var, laneMask, width) #define __shfl_xor_sync(mask, var, laneMask, width) __shfl_xor(var, laneMask, width)
#define __all_sync(mask, var) __all(var) #define __all_sync(mask, var) __all(var)
#define __any_sync(mask, var) __any(var) // #define __any_sync(mask, var) __any(var)
#define cublasStrsmBatched hipblasStrsmBatched #define cublasStrsmBatched hipblasStrsmBatched
#define cublasCreate hipblasCreate #define cublasCreate hipblasCreate
#define cublasDestroy hipblasDestroy #define cublasDestroy hipblasDestroy