From d3a74940b3348cbf30a373456ca43a774bc9723e Mon Sep 17 00:00:00 2001 From: Deven Desai Date: Wed, 7 Aug 2019 20:19:09 +0000 Subject: [PATCH] Updating that parallel_gpu_execute.sh script such that it works as expected for AMD GPus --- tensorflow/tools/ci_build/gpu_build/parallel_gpu_execute.sh | 1 + 1 file changed, 1 insertion(+) diff --git a/tensorflow/tools/ci_build/gpu_build/parallel_gpu_execute.sh b/tensorflow/tools/ci_build/gpu_build/parallel_gpu_execute.sh index 9c6390070c0..ee70f2f608b 100755 --- a/tensorflow/tools/ci_build/gpu_build/parallel_gpu_execute.sh +++ b/tensorflow/tools/ci_build/gpu_build/parallel_gpu_execute.sh @@ -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 # single command. export CUDA_VISIBLE_DEVICES=$i + export HIP_VISIBLE_DEVICES=$i echo "Running test $TEST_BINARY $* on GPU $CUDA_VISIBLE_DEVICES" "$TEST_BINARY" $@ )