From 1970d8c9e84b88c3c47bb6f5a0fa5e2855428122 Mon Sep 17 00:00:00 2001 From: Concedo <39025047+LostRuins@users.noreply.github.com> Date: Thu, 12 Jun 2025 16:44:46 +0800 Subject: [PATCH] uvos said it might work --- Makefile | 6 +++--- koboldcpp.sh | 10 +++++----- 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/Makefile b/Makefile index 7ac32a87b..a5736c920 100644 --- a/Makefile +++ b/Makefile @@ -260,15 +260,15 @@ endif HCC := $(ROCM_PATH)/llvm/bin/clang HCXX := $(ROCM_PATH)/llvm/bin/clang++ endif -ifdef LLAMA_ALLOW_WMMA +ifdef LLAMA_NO_WMMA + HIPFLAGS += -DGGML_HIP_NO_ROCWMMA_FATTN +else DETECT_ROCWMMA := $(shell find -L /opt/rocm/include /usr/include -type f -name rocwmma.hpp 2>/dev/null | head -n 1) ifdef DETECT_ROCWMMA HIPFLAGS += -DGGML_HIP_ROCWMMA_FATTN -I$(dir $(DETECT_ROCWMMA)) else HIPFLAGS += -DGGML_HIP_NO_ROCWMMA_FATTN endif -else - HIPFLAGS += -DGGML_HIP_NO_ROCWMMA_FATTN endif HIPFLAGS += -DGGML_USE_HIP -DGGML_HIP_NO_VMM -DGGML_USE_CUDA -DSD_USE_CUDA $(shell $(ROCM_PATH)/bin/hipconfig -C) diff --git a/koboldcpp.sh b/koboldcpp.sh index c0d0dcaef..b69527489 100755 --- a/koboldcpp.sh +++ b/koboldcpp.sh @@ -29,7 +29,7 @@ KCPP_CUDAAPPEND=-cuda${KCPP_CUDA//.}$KCPP_APPEND LLAMA_NOAVX2_FLAG="" ARCHES_FLAG="" -ALLOW_WMMA_FLAG="" +NO_WMMA_FLAG="" if [ -n "$NOAVX2" ]; then LLAMA_NOAVX2_FLAG="LLAMA_NOAVX2=1" fi @@ -39,14 +39,14 @@ fi if [ -n "$ARCHES_CU12" ]; then ARCHES_FLAG="LLAMA_ARCHES_CU12=1" fi -if [ -n "$ALLOW_WMMA" ]; then - ALLOW_WMMA_FLAG="LLAMA_ALLOW_WMMA=1" +if [ -n "$NO_WMMA" ]; then + NO_WMMA_FLAG="LLAMA_NO_WMMA=1" fi if [ "$KCPP_CUDA" = "rocm" ]; then - 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 + 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 else - 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 + 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 fi if [ $? -ne 0 ]; then