Skip to content

Commit 0fe6384

Browse files
committed
fix makefile
1 parent 605560d commit 0fe6384

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

Makefile

+2-2
Original file line numberDiff line numberDiff line change
@@ -140,8 +140,8 @@ ifdef LLAMA_HIPBLAS
140140
CC := $(ROCM_PATH)/llvm/bin/clang
141141
CXX := $(ROCM_PATH)/llvm/bin/clang++
142142
GPU_TARGETS = gfx900 gfx906 gfx908 gfx90a gfx1030
143-
CFLAGS += -DGGML_USE_HIPBLAS $(shell $(ROCM_PATH)/bin/hipconfig -C)
144-
CXXFLAGS += -DGGML_USE_HIPBLAS $(shell $(ROCM_PATH)/bin/hipconfig -C)
143+
CFLAGS += -DGGML_USE_HIPBLAS -DGGML_USE_CUBLAS $(shell $(ROCM_PATH)/bin/hipconfig -C)
144+
CXXFLAGS += -DGGML_USE_HIPBLAS -DGGML_USE_CUBLAS $(shell $(ROCM_PATH)/bin/hipconfig -C)
145145
LDFLAGS += -L/opt/rocm/lib -Wl,-rpath=$(ROCM_PATH)/lib -lhipblas -lamdhip64
146146
OBJS += ggml-cuda.o
147147
ggml-cuda.o: CXXFLAGS += $(addprefix --offload-arch=,$(GPU_TARGETS))

0 commit comments

Comments
 (0)