Skip to content

Commit

Permalink
Update cuda.yml
Browse files Browse the repository at this point in the history
  • Loading branch information
StrikerRUS authored Jan 7, 2021
1 parent 6cd014d commit bcc3f29
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions .github/workflows/cuda.yml
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,6 @@ jobs:
submodules: true
- name: Test CUDA
run: |
export ROOT_DOCKER_FOLDER=/LightGBM
cat > docker-script.sh <<EOF
nvidia-smi
apt-get update
Expand All @@ -52,7 +51,7 @@ jobs:
./cmake.sh --prefix=/usr/local --exclude-subdir
sed -i'.bak' 's/std::string device_type = "cpu";/std::string device_type = "cuda";/' /LightGBM/include/LightGBM/config.h
grep -q 'std::string device_type = "cuda"' /LightGBM/include/LightGBM/config.h || exit -1 # make sure that changes were really done
echo "Check sum of cuda_tree_learner.cpp"
echo "Check sum of cuda_tree_learner.cpp:"
sum /LightGBM/src/treelearner/cuda_tree_learner.cpp
mkdir /LightGBM/build && cd /LightGBM/build
cmake -DUSE_CUDA=ON ..
Expand Down

0 comments on commit bcc3f29

Please sign in to comment.