pkgs/development/python-modules/bitsandbytes/find-rocm-deps-with-cmake.patch at r-updates

tjh.dev / nixpkgs

fork atom

nixpkgs mirror (for testing) github.com/NixOS/nixpkgs

nix

fork atom

nixpkgs / pkgs / development / python-modules / bitsandbytes / find-rocm-deps-with-cmake.patch

at r-updates 23 lines 1.3 kB view raw

wrap content

 1diff --git a/CMakeLists.txt b/CMakeLists.txt
 2index 9c133e0..1aa8a53 100644
 3--- a/CMakeLists.txt
 4+++ b/CMakeLists.txt
 5@@ -285,6 +285,9 @@ if(BUILD_HIP)
 6     find_package_and_print_version(hipblas REQUIRED)
 7     find_package_and_print_version(hiprand REQUIRED)
 8     find_package_and_print_version(hipsparse REQUIRED)
 9+    find_package_and_print_version(rocblas REQUIRED)
10+    find_package_and_print_version(hip REQUIRED)
11+    find_package_and_print_version(hipcub REQUIRED)
12 
13     ## hacky way of excluding hip::amdhip64 (with it linked many tests unexpectedly fail e.g. adam8bit because of inaccuracies)
14     set_target_properties(hip::host PROPERTIES INTERFACE_LINK_LIBRARIES "")
15@@ -293,7 +296,7 @@ if(BUILD_HIP)
16 
17     target_include_directories(bitsandbytes PRIVATE ${CMAKE_SOURCE_DIR} ${CMAKE_SOURCE_DIR}/include ${ROCM_PATH}/include /include)
18     target_link_directories(bitsandbytes PRIVATE ${ROCM_PATH}/lib /lib)
19-    target_link_libraries(bitsandbytes PUBLIC roc::hipblas hip::hiprand roc::hipsparse)
20+    target_link_libraries(bitsandbytes PUBLIC roc::hipblas hip::hiprand roc::hipsparse roc::rocblas hip::device hip::hipcub)
21 
22     target_compile_definitions(bitsandbytes PUBLIC BNB_USE_HIP)
23     set_source_files_properties(${HIP_FILES} PROPERTIES LANGUAGE HIP)