File tree 2 files changed +3
-2
lines changed
2 files changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -359,6 +359,7 @@ if (LLAMA_HIPBLAS)
359
359
360
360
find_package (hip)
361
361
find_package (hipblas)
362
+ find_package (rocblas)
362
363
363
364
if (${hipblas_FOUND} AND ${hip_FOUND} )
364
365
message (STATUS "HIP and hipBLAS found" )
@@ -369,7 +370,7 @@ if (LLAMA_HIPBLAS)
369
370
target_compile_definitions (ggml-rocm PRIVATE K_QUANTS_PER_ITERATION=${LLAMA_CUDA_KQUANTS_ITER} )
370
371
target_compile_definitions (ggml-rocm PRIVATE GGML_CUDA_FORCE_DMMV)
371
372
set_source_files_properties (ggml-cuda.cu PROPERTIES LANGUAGE CXX)
372
- target_link_libraries (ggml-rocm PRIVATE hip::device PUBLIC hip::host roc::hipblas)
373
+ target_link_libraries (ggml-rocm PRIVATE hip::device PUBLIC hip::host roc::rocblas roc:: hipblas)
373
374
374
375
if (LLAMA_STATIC)
375
376
message (FATAL_ERROR "Static linking not supported for HIP/ROCm" )
Original file line number Diff line number Diff line change @@ -228,7 +228,7 @@ ifdef LLAMA_HIPBLAS
228
228
LLAMA_CUDA_KQUANTS_ITER ?= 2
229
229
CFLAGS += -DGGML_USE_HIPBLAS -DGGML_USE_CUBLAS $(shell $(ROCM_PATH)/bin/hipconfig -C)
230
230
CXXFLAGS += -DGGML_USE_HIPBLAS -DGGML_USE_CUBLAS $(shell $(ROCM_PATH)/bin/hipconfig -C)
231
- LDFLAGS += -L/opt/rocm/lib -Wl,-rpath=$(ROCM_PATH)/lib -lhipblas -lamdhip64
231
+ LDFLAGS += -L/opt/rocm/lib -Wl,-rpath=$(ROCM_PATH)/lib -lhipblas -lamdhip64 -lrocblas
232
232
OBJS += ggml-cuda.o
233
233
ggml-cuda.o : CXXFLAGS += $(addprefix --offload-arch=,$(GPU_TARGETS ) )
234
234
ggml-cuda.o : CXXFLAGS += -DGGML_CUDA_DMMV_X=$(LLAMA_CUDA_DMMV_X )
You can’t perform that action at this time.
0 commit comments