Merge pull request #2746 from ROCm/r2.15-rocm-enhanced-winograd_off

skipping winograd in single gpu test
ROCm · Nov 7, 2024 · 6321664 · 6321664
2 parents 5ee3d45 + bdefe4e
commit 6321664
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tensorflow/tools/ci_build/linux/rocm/run_gpu_single.sh b/tensorflow/tools/ci_build/linux/rocm/run_gpu_single.sh
@@ -72,7 +72,8 @@ if [ -f /usertools/rocm.bazelrc ]; then
              --action_env=TF_PYTHON_VERSION=$PYTHON_VERSION \
              --action_env=TF_ENABLE_ONEDNN_OPTS=0 \
              --test_env=TF_TESTS_PER_GPU=$TF_TESTS_PER_GPU \
-             --test_env=TF_GPU_COUNT=$TF_GPU_COUNT
+             --test_env=TF_GPU_COUNT=$TF_GPU_COUNT \
+             --test_env=MIOPEN_DEBUG_CONV_WINOGRAD=0
 else
 	# Legacy style: run configure then build
 	yes "" | $PYTHON_BIN_PATH configure.py
@@ -90,6 +91,7 @@ else
 	      --test_env=TF_TESTS_PER_GPU=$TF_TESTS_PER_GPU \
 	      --test_env=HSA_TOOLS_LIB=libroctracer64.so \
 	      --test_env=TF_PYTHON_VERSION=$PYTHON_VERSION \
+        --test_env=MIOPEN_DEBUG_CONV_WINOGRAD=0 \
 	      --action_env=OPENBLAS_CORETYPE=Haswell \
               --action_env=TF_ENABLE_ONEDNN_OPTS=0 \
 	      --test_timeout 920,2400,7200,9600 \