experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Updating that parallel_gpu_execute.sh script such that it works as expected for AMD GPus

Browse Source
This commit is contained in:
Deven Desai 2019-08-07 20:19:09 +00:00
parent 9c3a30dfff
commit d3a74940b3
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/tools/ci_build/gpu_build/parallel_gpu_execute.sh
View File

@ -68,6 +68,7 @@ for j in `seq 0 $((TF_TESTS_PER_GPU-1))`; do
# This export only works within the brackets, so it is isolated to one # This export only works within the brackets, so it is isolated to one
# single command. # single command.
export CUDA_VISIBLE_DEVICES=$i export CUDA_VISIBLE_DEVICES=$i
export HIP_VISIBLE_DEVICES=$i
echo "Running test $TEST_BINARY $* on GPU $CUDA_VISIBLE_DEVICES" echo "Running test $TEST_BINARY $* on GPU $CUDA_VISIBLE_DEVICES"
"$TEST_BINARY" $@ "$TEST_BINARY" $@
) )