Просмотр исходного кода

HIP: disable sync warp shuffel operators from clr amd_warp_sync_functions.h (#15273)

uvos 5 месяцев назад
Родитель
Сommit
b0493156fa
2 измененных файлов с 1 добавлено и 2 удалено
  1. 0 1
      ggml/src/ggml-cuda/fattn-wmma-f16.cu
  2. 1 1
      ggml/src/ggml-cuda/vendors/hip.h

+ 0 - 1
ggml/src/ggml-cuda/fattn-wmma-f16.cu

@@ -15,7 +15,6 @@ namespace wmma = mtmusa::wmma;
 namespace wmma = nvcuda::wmma;
 #endif // GGML_USE_MUSA
 #elif defined(GGML_HIP_ROCWMMA_FATTN) && defined(FP16_MMA_AVAILABLE)
-#undef HIP_ENABLE_WARP_SYNC_BUILTINS // conflicts with rocWMMA headers
 #include <rocwmma/rocwmma.hpp>
 namespace wmma = rocwmma;
 #endif // !defined(GGML_USE_HIP)

+ 1 - 1
ggml/src/ggml-cuda/vendors/hip.h

@@ -1,6 +1,6 @@
 #pragma once
 
-#define HIP_ENABLE_WARP_SYNC_BUILTINS 1
+#define HIP_DISABLE_WARP_SYNC_BUILTINS 1
 #include <hip/hip_runtime.h>
 #include <hipblas/hipblas.h>
 #include <hip/hip_fp16.h>