update CI (#375)

dmlc · Oct 24, 2018 · 06ee6cf · 06ee6cf
1 parent e1a61f5
commit 06ee6cf
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/ci/step_documentation.sh b/ci/step_documentation.sh
@@ -1,10 +1,9 @@
 #!/bin/bash
-gpu=$(($EXECUTOR_NUMBER % 4))
 set -ex
 export LD_LIBRARY_PATH=/usr/local/cuda/lib64
 source ci/prepare_clean_env.sh doc
 ci/install_dep.sh
-export CUDA_VISIBLE_DEVICES=${gpu}
+export CUDA_VISIBLE_DEVICES=$EXECUTOR_NUMBER
 make clean_doc
 make docs
 set +ex
diff --git a/ci/step_unit_test.sh b/ci/step_unit_test.sh
@@ -2,12 +2,11 @@
 lang=$1
 branch=$2
 capture_flag=$3
-gpu=$(($EXECUTOR_NUMBER % 4))
 set -ex
 export LD_LIBRARY_PATH=/usr/local/cuda/lib64
 source ci/prepare_clean_env.sh ${lang}
 ci/install_dep.sh
-export CUDA_VISIBLE_DEVICES=${gpu}
+export CUDA_VISIBLE_DEVICES=$EXECUTOR_NUMBER
 py.test -v ${capture_flag} -n 4 -m "not serial" --durations=50 --cov=gluonnlp tests/unittest
 py.test -v ${capture_flag} -n 0 -m "serial" --durations=50 --cov=gluonnlp tests/unittest
 ci/codecov.sh -c -F ${branch},${lang} -n unittests