From 83aef0b1479685f492adf7f918c1050ca400efb1 Mon Sep 17 00:00:00 2001 From: Zachary Streeter <90640993+zstreet87@users.noreply.github.com> Date: Thu, 7 Nov 2024 11:25:44 -0600 Subject: [PATCH] skipping winograd in single gpu test (#2747) --- tensorflow/tools/ci_build/linux/rocm/run_gpu_single.sh | 4 +++- 1 file changed, 3 insertions(+), 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 index 94e61660c237d3..f084859d7c777b 100755 --- a/tensorflow/tools/ci_build/linux/rocm/run_gpu_single.sh +++ b/tensorflow/tools/ci_build/linux/rocm/run_gpu_single.sh @@ -69,7 +69,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 @@ -85,6 +86,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 \