Skip to content

Commit 947e75c

Browse files
committed
change default GPUs and add ALL_AMD_GPU flag
Removed workstation GPUs being built by default. Added a flag to build all AMD ROCm GPUs, use it like "make LLAMA_HIPBLAS=1 ALL_AMD_GPU=1"
1 parent 56c1218 commit 947e75c

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

Makefile

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -214,7 +214,11 @@ ifdef LLAMA_HIPBLAS
214214
ROCM_PATH ?= /opt/rocm
215215
HCC := $(ROCM_PATH)/llvm/bin/clang
216216
HCXX := $(ROCM_PATH)/llvm/bin/clang++
217-
GPU_TARGETS ?= gfx803 gfx900 gfx906 gfx908 gfx90a gfx1030 gfx1100 $(shell $(ROCM_PATH)/llvm/bin/amdgpu-arch)
217+
ifdef ALL_AMD_GPU # Build for all AMD GPU types
218+
GPU_TARGETS ?= gfx803 gfx900 gfx906 gfx908 gfx90a gfx1010 gfx1030 gfx1031 gfx1032 gfx1100 gfx1101 gfx1102 $(shell $(ROCM_PATH)/llvm/bin/amdgpu-arch)
219+
else
220+
GPU_TARGETS ?= gfx803 gfx900 gfx1030 gfx1100 $(shell $(ROCM_PATH)/llvm/bin/amdgpu-arch) # Only build for most common + currently active GPU
221+
endif
218222
LLAMA_CUDA_DMMV_X ?= 32
219223
LLAMA_CUDA_MMV_Y ?= 1
220224
LLAMA_CUDA_KQUANTS_ITER ?= 2
@@ -296,6 +300,8 @@ ifeq ($(OS),Windows_NT)
296300
endif
297301
else
298302
DEFAULT_BUILD = $(CXX) $(CXXFLAGS) $^ -shared -o $@.so $(LDFLAGS)
303+
FAILSAFE_BUILD = $(CXX) $(CXXFLAGS) $^ -shared -o $@.so $(LDFLAGS)
304+
NOAVX2_BUILD = $(CXX) $(CXXFLAGS) $^ -shared -o $@.so $(LDFLAGS)
299305

300306
ifdef LLAMA_OPENBLAS
301307
OPENBLAS_BUILD = $(CXX) $(CXXFLAGS) $^ $(ARCH_ADD) -lopenblas -shared -o $@.so $(LDFLAGS)
@@ -521,10 +527,5 @@ llama-bench: examples/llama-bench/llama-bench.cpp build-info.h ggml_cublas.o ggm
521527
simpleclinfo: simpleclinfo.cpp
522528
$(CXX) $(CXXFLAGS) $^ lib/OpenCL.lib lib/clblast.lib -o $@ $(LDFLAGS)
523529

524-
#window simple clinfo
525-
simpleclinfo: simpleclinfo.cpp
526-
$(CXX) $(CXXFLAGS) $^ lib/OpenCL.lib lib/clblast.lib -o $@ $(LDFLAGS)
527-
528-
529530
build-info.h:
530531
$(DONOTHING)

0 commit comments

Comments
 (0)