File tree 1 file changed +11
-4
lines changed
1 file changed +11
-4
lines changed Original file line number Diff line number Diff line change @@ -198,10 +198,17 @@ ggml_v3-cuda.o: otherarch/ggml_v3-cuda.cu otherarch/ggml_v3-cuda.h
198
198
endif # LLAMA_CUBLAS
199
199
200
200
ifdef LLAMA_HIPBLAS
201
- ROCM_PATH ?= /opt/rocm
202
- HCC := $(ROCM_PATH)/llvm/bin/clang
203
- HCXX := $(ROCM_PATH)/llvm/bin/clang++
204
- GPU_TARGETS ?= gfx803 gfx900 gfx906 gfx908 gfx90a gfx1030 gfx1100 $(shell $(ROCM_PATH)/llvm/bin/amdgpu-arch)
201
+ ifeq ($(wildcard /opt/rocm),)
202
+ ROCM_PATH ?= /usr
203
+ GPU_TARGETS ?= $(shell $(shell which amdgpu-arch))
204
+ HCC := $(ROCM_PATH)/bin/hipcc
205
+ HCXX := $(ROCM_PATH)/bin/hipcc
206
+ else
207
+ ROCM_PATH ?= /opt/rocm
208
+ GPU_TARGETS ?= gfx803 gfx900 gfx906 gfx908 gfx90a gfx1030 gfx1100 $(shell $(ROCM_PATH)/llvm/bin/amdgpu-arch)
209
+ HCC := $(ROCM_PATH)/llvm/bin/clang
210
+ HCXX := $(ROCM_PATH)/llvm/bin/clang++
211
+ endif
205
212
LLAMA_CUDA_DMMV_X ?= 32
206
213
LLAMA_CUDA_MMV_Y ?= 1
207
214
LLAMA_CUDA_KQUANTS_ITER ?= 2
You can’t perform that action at this time.
0 commit comments