Explorar el Código

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

uvos hace 5 meses
padre
commit
b0493156fa
Se han modificado 2 ficheros con 1 adiciones y 2 borrados
  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;
 namespace wmma = nvcuda::wmma;
 #endif // GGML_USE_MUSA
 #endif // GGML_USE_MUSA
 #elif defined(GGML_HIP_ROCWMMA_FATTN) && defined(FP16_MMA_AVAILABLE)
 #elif defined(GGML_HIP_ROCWMMA_FATTN) && defined(FP16_MMA_AVAILABLE)
-#undef HIP_ENABLE_WARP_SYNC_BUILTINS // conflicts with rocWMMA headers
 #include <rocwmma/rocwmma.hpp>
 #include <rocwmma/rocwmma.hpp>
 namespace wmma = rocwmma;
 namespace wmma = rocwmma;
 #endif // !defined(GGML_USE_HIP)
 #endif // !defined(GGML_USE_HIP)

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

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