Skip to content

Commit 54a63c1

Browse files
committed
Update Makefile for the Cuda kernels
1 parent 0fd8363 commit 54a63c1

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

Makefile

+9-2
Original file line numberDiff line numberDiff line change
@@ -108,8 +108,15 @@ ggml-cuda.o: ggml-cuda.cu ggml-cuda.h
108108
nvcc -arch=native -c -o $@ $<
109109
endif
110110
ifdef LLAMA_HIPBLAS
111-
CFLAGS += -DGGML_USE_HIPBLAS -D__HIP_PLATFORM_AMD__ -I/opt/rocm/include
112-
LDFLAGS += -lhipblas -lamdhip64 -L/opt/rocm/lib
111+
ROCMPATH?= /opt/rocm
112+
CFLAGS += -DGGML_USE_HIPBLAS -D__HIP_PLATFORM_AMD__ -I$(ROCMPATH)/include
113+
CXXFLAGS+= -D__HIP_PLATFORM_AMD__ -I$(ROCMPATH)/include
114+
HIPFLAGS?= -amdgpu-early-inline-all=true -amdgpu-function-calls=false -march=native
115+
LDFLAGS += -lhipblas -lamdhip64 -L$(ROCMPATH)/lib
116+
HIPCC ?= $(ROCMPATH)/bin/hipcc
117+
OBJS += ggml-cuda.o
118+
ggml-cuda.o: ggml-cuda.cu ggml-cuda.h
119+
$(HIPCC) $(CXXFLAGS) -x hip $(HIPFLAGS) -c -o $@ $<
113120
endif
114121
ifdef LLAMA_GPROF
115122
CFLAGS += -pg

0 commit comments

Comments
 (0)