From 226c3dc98ad785be149b705aca99e002171a1bde Mon Sep 17 00:00:00 2001 From: baonudesifeizhai <85092850+baonudesifeizhai@users.noreply.github.com> Date: Tue, 23 Dec 2025 15:18:39 -0500 Subject: [PATCH] Update tests/compile/fullgraph/test_full_graph.py MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: Luka Govedič Signed-off-by: baonudesifeizhai <85092850+baonudesifeizhai@users.noreply.github.com> --- tests/compile/fullgraph/test_full_graph.py | 20 -------------------- 1 file changed, 20 deletions(-) diff --git a/tests/compile/fullgraph/test_full_graph.py b/tests/compile/fullgraph/test_full_graph.py index 87a2f37af540f..5a3ed71d82df6 100644 --- a/tests/compile/fullgraph/test_full_graph.py +++ b/tests/compile/fullgraph/test_full_graph.py @@ -195,26 +195,6 @@ def test_custom_compile_config( ): pytest.skip("inductor graph partition is only available in PyTorch 2.9+") - # For get_raw_stream patch test: verify version and log context - if compilation_config.compile_sizes and len(compilation_config.compile_sizes) > 1: - # Verify torch version >= 2.9.0 for compile_sizes autotune - if not is_torch_equal_or_newer("2.9.0"): - pytest.skip( - f"compile_sizes autotune requires torch >= 2.9.0, " - f"got {torch.__version__}" - ) - - # Log version context for get_raw_stream patch testing - is_torch_2_9 = is_torch_equal("2.9.0") or is_torch_equal("2.9.1") - version_context = ( - "2.9.x (get_raw_stream patch applied)" - if is_torch_2_9 - else "2.10+ (get_raw_stream patch not needed)" - ) - print( - f"Testing compile_sizes with torch {torch.__version__} ({version_context})" - ) - print(f"MODEL={model}") run_model(compilation_config, model, **model_kwargs)