Skip to content

Commit afcb8fe

Browse files
committed
Add new config option
1 parent 8c2c497 commit afcb8fe

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

CMakeLists.txt

+3
Original file line numberDiff line numberDiff line change
@@ -367,6 +367,9 @@ if (LLAMA_HIPBLAS)
367367
target_compile_definitions(ggml-rocm PRIVATE GGML_CUDA_DMMV_X=${LLAMA_CUDA_DMMV_X})
368368
target_compile_definitions(ggml-rocm PRIVATE GGML_CUDA_DMMV_Y=${LLAMA_CUDA_DMMV_Y})
369369
target_compile_definitions(ggml-rocm PRIVATE K_QUANTS_PER_ITERATION=${LLAMA_CUDA_KQUANTS_ITER})
370+
if (LLAMA_CUDA_FORCE_DMMV)
371+
target_compile_definitions(ggml-rocm PRIVATE GGML_CUDA_FORCE_DMMV)
372+
endif()
370373
set_source_files_properties(ggml-cuda.cu PROPERTIES LANGUAGE CXX)
371374
target_link_libraries(ggml-rocm PRIVATE hip::device PUBLIC hip::host roc::hipblas)
372375

0 commit comments

Comments
 (0)