File tree 1 file changed +9
-2
lines changed
1 file changed +9
-2
lines changed Original file line number Diff line number Diff line change @@ -108,8 +108,15 @@ ggml-cuda.o: ggml-cuda.cu ggml-cuda.h
108
108
nvcc -arch=native -c -o $@ $<
109
109
endif
110
110
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 $@ $<
113
120
endif
114
121
ifdef LLAMA_GPROF
115
122
CFLAGS += -pg
You can’t perform that action at this time.
0 commit comments