File tree Expand file tree Collapse file tree 2 files changed +8
-8
lines changed Expand file tree Collapse file tree 2 files changed +8
-8
lines changed Original file line number Diff line number Diff line change @@ -260,15 +260,15 @@ endif
260260 HCC := $(ROCM_PATH)/llvm/bin/clang
261261 HCXX := $(ROCM_PATH)/llvm/bin/clang++
262262endif
263- ifdef LLAMA_ALLOW_WMMA
263+ ifdef LLAMA_NO_WMMA
264+ HIPFLAGS += -DGGML_HIP_NO_ROCWMMA_FATTN
265+ else
264266 DETECT_ROCWMMA := $(shell find -L /opt/rocm/include /usr/include -type f -name rocwmma.hpp 2>/dev/null | head -n 1)
265267ifdef DETECT_ROCWMMA
266268 HIPFLAGS += -DGGML_HIP_ROCWMMA_FATTN -I$(dir $(DETECT_ROCWMMA))
267269else
268270 HIPFLAGS += -DGGML_HIP_NO_ROCWMMA_FATTN
269271endif
270- else
271- HIPFLAGS += -DGGML_HIP_NO_ROCWMMA_FATTN
272272endif
273273
274274 HIPFLAGS += -DGGML_USE_HIP -DGGML_HIP_NO_VMM -DGGML_USE_CUDA -DSD_USE_CUDA $(shell $(ROCM_PATH)/bin/hipconfig -C)
Original file line number Diff line number Diff line change @@ -29,7 +29,7 @@ KCPP_CUDAAPPEND=-cuda${KCPP_CUDA//.}$KCPP_APPEND
2929
3030LLAMA_NOAVX2_FLAG=" "
3131ARCHES_FLAG=" "
32- ALLOW_WMMA_FLAG =" "
32+ NO_WMMA_FLAG =" "
3333if [ -n " $NOAVX2 " ]; then
3434 LLAMA_NOAVX2_FLAG=" LLAMA_NOAVX2=1"
3535fi
3939if [ -n " $ARCHES_CU12 " ]; then
4040 ARCHES_FLAG=" LLAMA_ARCHES_CU12=1"
4141fi
42- if [ -n " $ALLOW_WMMA " ]; then
43- ALLOW_WMMA_FLAG= " LLAMA_ALLOW_WMMA =1"
42+ if [ -n " $NO_WMMA " ]; then
43+ NO_WMMA_FLAG= " LLAMA_NO_WMMA =1"
4444fi
4545
4646if [ " $KCPP_CUDA " = " rocm" ]; then
47- bin/micromamba run -r conda -p conda/envs/linux make -j$( nproc) LLAMA_VULKAN=1 LLAMA_CLBLAST=1 LLAMA_HIPBLAS=1 LLAMA_PORTABLE=1 LLAMA_USE_BUNDLED_GLSLC=1 LLAMA_ADD_CONDA_PATHS=1 $LLAMA_NOAVX2_FLAG $ARCHES_FLAG $ALLOW_WMMA_FLAG
47+ bin/micromamba run -r conda -p conda/envs/linux make -j$( nproc) LLAMA_VULKAN=1 LLAMA_CLBLAST=1 LLAMA_HIPBLAS=1 LLAMA_PORTABLE=1 LLAMA_USE_BUNDLED_GLSLC=1 LLAMA_ADD_CONDA_PATHS=1 $LLAMA_NOAVX2_FLAG $ARCHES_FLAG $NO_WMMA_FLAG
4848else
49- bin/micromamba run -r conda -p conda/envs/linux make -j$( nproc) LLAMA_VULKAN=1 LLAMA_CLBLAST=1 LLAMA_CUBLAS=1 LLAMA_PORTABLE=1 LLAMA_USE_BUNDLED_GLSLC=1 LLAMA_ADD_CONDA_PATHS=1 $LLAMA_NOAVX2_FLAG $ARCHES_FLAG $ALLOW_WMMA_FLAG
49+ bin/micromamba run -r conda -p conda/envs/linux make -j$( nproc) LLAMA_VULKAN=1 LLAMA_CLBLAST=1 LLAMA_CUBLAS=1 LLAMA_PORTABLE=1 LLAMA_USE_BUNDLED_GLSLC=1 LLAMA_ADD_CONDA_PATHS=1 $LLAMA_NOAVX2_FLAG $ARCHES_FLAG $NO_WMMA_FLAG
5050fi
5151
5252if [ $? -ne 0 ]; then
You can’t perform that action at this time.
0 commit comments