From 915c7613bb36628f27255cd40371d8c47b1912b3 Mon Sep 17 00:00:00 2001 From: Zachary Streeter <90640993+zstreet87@users.noreply.github.com> Date: Thu, 7 Nov 2024 11:26:04 -0600 Subject: [PATCH] skipping winograd in single gpu test (#2748) --- 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 f27532935f37e9..7c7e710396bca5 100755 --- a/tensorflow/tools/ci_build/linux/rocm/run_gpu_single.sh +++ b/tensorflow/tools/ci_build/linux/rocm/run_gpu_single.sh @@ -71,7 +71,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 @@ -89,6 +90,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 \