Skip to content

GitLab

"src/git@developer.sourcefind.cn:tianlh/lightgbm-dcu.git" did not exist on "db3d1f89cae48833e4da361b0b403bb85ba1cfea"
Commit 3eafff74 authored by Jeff Daily's avatar Jeff Daily
Browse files

add --use-rocm option to build-python.sh

parent 20996c92
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 0 deletions
build-python.sh
View file @ 3eafff74
...@@ -38,6 +38,8 @@ ...@@ -38,6 +38,8 @@
# Compile 32-bit version. # Compile 32-bit version.
# --cuda # --cuda
# Compile CUDA version. # Compile CUDA version.
# --rocm
# Compile CUDA version.
# --gpu # --gpu
# Compile GPU version. # Compile GPU version.
# --integrated-opencl # --integrated-opencl
...@@ -142,6 +144,9 @@ while [ $# -gt 0 ]; do ...@@ -142,6 +144,9 @@ while [ $# -gt 0 ]; do
--cuda) --cuda)
BUILD_ARGS="${BUILD_ARGS} --config-setting=cmake.define.USE_CUDA=ON" BUILD_ARGS="${BUILD_ARGS} --config-setting=cmake.define.USE_CUDA=ON"
;; ;;
--rocm)
BUILD_ARGS="${BUILD_ARGS} --config-setting=cmake.define.USE_ROCM=ON"
;;
--gpu) --gpu)
BUILD_ARGS="${BUILD_ARGS} --config-setting=cmake.define.USE_GPU=ON" BUILD_ARGS="${BUILD_ARGS} --config-setting=cmake.define.USE_GPU=ON"
;; ;;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment