Skip to content

Commit 1970d8c

Browse files
committed
uvos said it might work
1 parent 5cdb2d3 commit 1970d8c

File tree

2 files changed

+8
-8
lines changed

2 files changed

+8
-8
lines changed

Makefile

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -260,15 +260,15 @@ endif
260260
HCC := $(ROCM_PATH)/llvm/bin/clang
261261
HCXX := $(ROCM_PATH)/llvm/bin/clang++
262262
endif
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)
265267
ifdef DETECT_ROCWMMA
266268
HIPFLAGS += -DGGML_HIP_ROCWMMA_FATTN -I$(dir $(DETECT_ROCWMMA))
267269
else
268270
HIPFLAGS += -DGGML_HIP_NO_ROCWMMA_FATTN
269271
endif
270-
else
271-
HIPFLAGS += -DGGML_HIP_NO_ROCWMMA_FATTN
272272
endif
273273

274274
HIPFLAGS += -DGGML_USE_HIP -DGGML_HIP_NO_VMM -DGGML_USE_CUDA -DSD_USE_CUDA $(shell $(ROCM_PATH)/bin/hipconfig -C)

koboldcpp.sh

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ KCPP_CUDAAPPEND=-cuda${KCPP_CUDA//.}$KCPP_APPEND
2929

3030
LLAMA_NOAVX2_FLAG=""
3131
ARCHES_FLAG=""
32-
ALLOW_WMMA_FLAG=""
32+
NO_WMMA_FLAG=""
3333
if [ -n "$NOAVX2" ]; then
3434
LLAMA_NOAVX2_FLAG="LLAMA_NOAVX2=1"
3535
fi
@@ -39,14 +39,14 @@ fi
3939
if [ -n "$ARCHES_CU12" ]; then
4040
ARCHES_FLAG="LLAMA_ARCHES_CU12=1"
4141
fi
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"
4444
fi
4545

4646
if [ "$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
4848
else
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
5050
fi
5151

5252
if [ $? -ne 0 ]; then

0 commit comments

Comments
 (0)