Skip to content

Commit a593a4f

Browse files
committed
Add missing parameters
1 parent 174bf6a commit a593a4f

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

CMakeLists.txt

+2
Original file line numberDiff line numberDiff line change
@@ -233,6 +233,8 @@ if (LLAMA_HIPBLAS)
233233
message(STATUS "HIP and hipBLAS found")
234234
add_compile_definitions(GGML_USE_HIPBLAS GGML_USE_CUBLAS)
235235
add_library(ggml-rocm OBJECT ggml-cuda.cu ggml-cuda.h)
236+
target_compile_definitions(ggml-rocm PRIVATE GGML_CUDA_DMMV_X=${LLAMA_CUDA_DMMV_X})
237+
target_compile_definitions(ggml-rocm PRIVATE GGML_CUDA_DMMV_Y=${LLAMA_CUDA_DMMV_Y})
236238
set_source_files_properties(ggml-cuda.cu PROPERTIES LANGUAGE CXX)
237239
target_link_libraries(ggml-rocm PRIVATE hip::device PUBLIC hip::host roc::hipblas)
238240

0 commit comments

Comments
 (0)