Richard Zou 9b6504c307

[BugFix] Work around graph partition x torch.compile cache issue (#26956) ...

Signed-off-by: Richard Zou <zou3519@gmail.com>

2025-10-15 20:06:11 -07:00

..

piecewise

[BugFix] Work around graph partition x torch.compile cache issue (#26956)

2025-10-15 20:06:11 -07:00

__init__.py

[torch.compile] register allreduce operations as custom ops (#8526)

2024-09-16 22:57:57 -07:00

backend.py

Update Optional[x] -> x | None and Union[x, y] to x | y (#26633)

2025-10-12 09:51:31 -07:00

silly_attention.py

[torch.compile] Fix tests for torch==2.9 inductor partition (#26116)

2025-10-14 19:55:02 -04:00

test_aot_compile.py

[Frontend][torch.compile] CompilationConfig Overhaul (#20283): name change compilation level to compilation mode, deprecation compilation level (#26355)

2025-10-15 02:51:16 +00:00

test_async_tp.py

[Frontend][torch.compile] CompilationConfig Overhaul (#20283): name change compilation level to compilation mode, deprecation compilation level (#26355)

2025-10-15 02:51:16 +00:00

test_basic_correctness.py

[Frontend][torch.compile] CompilationConfig Overhaul (#20283): name change compilation level to compilation mode, deprecation compilation level (#26355)

2025-10-15 02:51:16 +00:00

test_config.py

[Frontend][torch.compile] CompilationConfig Overhaul (#20283): name change compilation level to compilation mode, deprecation compilation level (#26355)

2025-10-15 02:51:16 +00:00

test_decorator.py

[Graph Partition] pass tests for decorator (#26831)

2025-10-15 06:39:48 +00:00

test_full_graph.py

[Frontend][torch.compile] CompilationConfig Overhaul (#20283): name change compilation level to compilation mode, deprecation compilation level (#26355)

2025-10-15 02:51:16 +00:00

test_functionalization.py

Fix per file ruff ignores related to line length (#26262)

2025-10-06 05:12:40 +00:00

test_fusion_all_reduce.py

[Frontend][torch.compile] CompilationConfig Overhaul (#20283): name change compilation level to compilation mode, deprecation compilation level (#26355)

2025-10-15 02:51:16 +00:00

test_fusion_attn.py

Move query quantization to attention layer for Flashinfer & Triton. (#26534)

2025-10-15 19:01:38 -04:00

test_fusion.py

[Frontend][torch.compile] CompilationConfig Overhaul (#20283): name change compilation level to compilation mode, deprecation compilation level (#26355)

2025-10-15 02:51:16 +00:00

test_noop_elimination.py

[Frontend][torch.compile] CompilationConfig Overhaul (#20283): name change compilation level to compilation mode, deprecation compilation level (#26355)

2025-10-15 02:51:16 +00:00

test_pass_manager.py

Convert formatting to use ruff instead of yapf + isort (#26247)

2025-10-05 07:06:22 -07:00

test_sequence_parallelism.py

[CI Failure] Fix tests with missing TinyLlama-1.1B-Chat-v1.0-FP8-e2e (#26816)

2025-10-14 18:38:59 +00:00

test_silu_mul_quant_fusion.py

Remove all references to yapf as it's no longer used (#26251)

2025-10-05 09:18:11 -07:00

test_wrapper.py

[Frontend][torch.compile] CompilationConfig Overhaul (#20283): name change compilation level to compilation mode, deprecation compilation level (#26355)

2025-10-15 02:51:16 +00:00