Public/koboldcpp
1
0
Fork 0
mirror of https://github.com/LostRuins/koboldcpp.git synced 2026-09-19 01:05:09 +02:00
Code Issues Packages Projects Releases Wiki Activity

Update Makefile for the Cuda kernels

Browse Source
This commit is contained in:
Henri Vasserman
2023-04-20 22:19:22 +03:00
parent 0fd8363adc
commit 54a63c10e8
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Makefile
+9 -2
View File
@@ -108,8 +108,15 @@ ggml-cuda.o: ggml-cuda.cu ggml-cuda.h
nvcc -arch=native -c -o $@ $<
endif
ifdef LLAMA_HIPBLAS
CFLAGS += -DGGML_USE_HIPBLAS -D__HIP_PLATFORM_AMD__ -I/opt/rocm/include
LDFLAGS += -lhipblas -lamdhip64 -L/opt/rocm/lib
ROCMPATH?= /opt/rocm
CFLAGS += -DGGML_USE_HIPBLAS -D__HIP_PLATFORM_AMD__ -I$(ROCMPATH)/include
CXXFLAGS+= -D__HIP_PLATFORM_AMD__ -I$(ROCMPATH)/include
HIPFLAGS?= -amdgpu-early-inline-all=true -amdgpu-function-calls=false -march=native
LDFLAGS += -lhipblas -lamdhip64 -L$(ROCMPATH)/lib
HIPCC ?= $(ROCMPATH)/bin/hipcc
OBJS += ggml-cuda.o
ggml-cuda.o: ggml-cuda.cu ggml-cuda.h
$(HIPCC) $(CXXFLAGS) -x hip $(HIPFLAGS) -c -o $@ $<
endif
ifdef LLAMA_GPROF
CFLAGS += -pg