Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[CI] Build pip wheel with RMM support (#9383)

Browse Source
This commit is contained in:
Philip Hyunsu Cho
2023-07-18 01:52:26 -07:00
committed by GitHub
parent 6e18d3a290
commit e082718c66
8 changed files with 64 additions and 70 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/ci_build/prune_libnccl.sh
View File

@@ -26,7 +26,7 @@ set_property(TARGET test PROPERTY CUDA_ARCHITECTURES \${CMAKE_CUDA_ARCHITECTURES
set(CMAKE_EXPORT_COMPILE_COMMANDS ON)
EOF
cmake . -GNinja
cmake . -GNinja -DCMAKE_EXPORT_COMPILE_COMMANDS=ON
gen_code=$(grep -o -- '--generate-code=\S*' compile_commands.json | paste -sd ' ')
nvprune ${gen_code} /usr/lib64/libnccl_static.a -o ../libnccl_static.a