Skip to content

Commit 9ab9045

Browse files
Debian Unstable compatibility for HIP (ggml-org#620)
* Support rocm on Debian unstable * Update Makefile
1 parent c939bbf commit 9ab9045

File tree

1 file changed

+11
-4
lines changed

1 file changed

+11
-4
lines changed

Makefile

+11-4
Original file line numberDiff line numberDiff line change
@@ -198,10 +198,17 @@ ggml_v3-cuda.o: otherarch/ggml_v3-cuda.cu otherarch/ggml_v3-cuda.h
198198
endif # LLAMA_CUBLAS
199199

200200
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
205212
LLAMA_CUDA_DMMV_X ?= 32
206213
LLAMA_CUDA_MMV_Y ?= 1
207214
LLAMA_CUDA_KQUANTS_ITER ?= 2

0 commit comments

Comments
 (0)