From a65c11a0786afd9480b79c5169ec41e321f2beaa Mon Sep 17 00:00:00 2001 From: erman-gurses Date: Tue, 10 Sep 2024 01:45:31 -0500 Subject: [PATCH] Revert env vars to default values Signed-off-by: erman-gurses --- tests/kernel/wave/wave_e2e_test.py | 2 +- tests/kernel/wave/wave_gemm_test.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/kernel/wave/wave_e2e_test.py b/tests/kernel/wave/wave_e2e_test.py index 70a3a348..3c07afef 100644 --- a/tests/kernel/wave/wave_e2e_test.py +++ b/tests/kernel/wave/wave_e2e_test.py @@ -9,7 +9,7 @@ import torch import json -_run_e2e = int(os.environ.get("WAVE_RUN_E2E_TESTS", 1)) +_run_e2e = int(os.environ.get("WAVE_RUN_E2E_TESTS", 0)) require_e2e = pytest.mark.skipif(not _run_e2e, reason="e2e tests are disabled") default_test_shapes = [(1, 128), (256, 64), (256, 128), (256, 256), (256, 1024)] diff --git a/tests/kernel/wave/wave_gemm_test.py b/tests/kernel/wave/wave_gemm_test.py index b0ccdbce..15af87d2 100644 --- a/tests/kernel/wave/wave_gemm_test.py +++ b/tests/kernel/wave/wave_gemm_test.py @@ -10,7 +10,7 @@ import os import json -_run_e2e = int(os.environ.get("WAVE_RUN_E2E_TESTS", 1)) +_run_e2e = int(os.environ.get("WAVE_RUN_E2E_TESTS", 0)) require_e2e = pytest.mark.skipif(not _run_e2e, reason="e2e tests are disabled") default_test_shapes = [(1024, 5120, 640), (2048, 10240, 1280), (4096, 20480, 2560)]