Something went wrong. Try again.
lol
Something went wrong. Try again.
1.3 kB · 23 lines
at master
123456789101112131415161718192021222324diff --git a/CMakeLists.txt b/CMakeLists.txtindex 9c133e0..1aa8a53 100644--- a/CMakeLists.txt+++ b/CMakeLists.txt@@ -285,6 +285,9 @@ if(BUILD_HIP) find_package_and_print_version(hipblas REQUIRED) find_package_and_print_version(hiprand REQUIRED) find_package_and_print_version(hipsparse REQUIRED)+ find_package_and_print_version(rocblas REQUIRED)+ find_package_and_print_version(hip REQUIRED)+ find_package_and_print_version(hipcub REQUIRED) ## hacky way of excluding hip::amdhip64 (with it linked many tests unexpectedly fail e.g. adam8bit because of inaccuracies) set_target_properties(hip::host PROPERTIES INTERFACE_LINK_LIBRARIES "")@@ -293,7 +296,7 @@ if(BUILD_HIP) target_include_directories(bitsandbytes PRIVATE ${CMAKE_SOURCE_DIR} ${CMAKE_SOURCE_DIR}/include ${ROCM_PATH}/include /include) target_link_directories(bitsandbytes PRIVATE ${ROCM_PATH}/lib /lib)- target_link_libraries(bitsandbytes PUBLIC roc::hipblas hip::hiprand roc::hipsparse)+ target_link_libraries(bitsandbytes PUBLIC roc::hipblas hip::hiprand roc::hipsparse roc::rocblas hip::device hip::hipcub) target_compile_definitions(bitsandbytes PUBLIC BNB_USE_HIP) set_source_files_properties(${HIP_FILES} PROPERTIES LANGUAGE HIP)