Revert env vars to default values

Signed-off-by: erman-gurses <[email protected]>
iree-org · Sep 10, 2024 · a65c11a · a65c11a
1 parent 14abe0b
commit a65c11a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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)]