Skip to content

Commit 60268de

Browse files
committed
update targets for rocm
1 parent f142221 commit 60268de

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

Makefile

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -242,12 +242,16 @@ endif # LLAMA_CUBLAS
242242
ifdef LLAMA_HIPBLAS
243243
ifeq ($(wildcard /opt/rocm),)
244244
ROCM_PATH ?= /usr
245+
ifdef LLAMA_PORTABLE
246+
GPU_TARGETS ?= gfx803 gfx900 gfx906 gfx908 gfx90a gfx1010 gfx1030 gfx1031 gfx1032 gfx1100 gfx1101 gfx1102 $(shell $(shell which amdgpu-arch))
247+
else
245248
GPU_TARGETS ?= $(shell $(shell which amdgpu-arch))
249+
endif
246250
HCC := $(ROCM_PATH)/bin/hipcc
247251
HCXX := $(ROCM_PATH)/bin/hipcc
248252
else
249253
ROCM_PATH ?= /opt/rocm
250-
GPU_TARGETS ?= gfx803 gfx900 gfx906 gfx908 gfx90a gfx1030 gfx1100 $(shell $(ROCM_PATH)/llvm/bin/amdgpu-arch)
254+
GPU_TARGETS ?= gfx803 gfx900 gfx906 gfx908 gfx90a gfx1010 gfx1030 gfx1031 gfx1032 gfx1100 gfx1101 gfx1102 $(shell $(ROCM_PATH)/llvm/bin/amdgpu-arch)
251255
HCC := $(ROCM_PATH)/llvm/bin/clang
252256
HCXX := $(ROCM_PATH)/llvm/bin/clang++
253257
endif

0 commit comments

Comments
 (0)