Gregory Shtrasberg f231e5bc21

[ROCm] Split AITER unified attention into its own backend (#25507) ...

Signed-off-by: Gregory Shtrasberg <Gregory.Shtrasberg@amd.com>

2025-10-06 22:49:23 +00:00

..

piecewise

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

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

__init__.py

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

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

backend.py

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

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

silly_attention.py

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

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

test_async_tp.py

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

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

test_basic_correctness.py

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

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

test_config.py

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

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

test_decorator.py

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

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

test_full_graph.py

Fix per file ruff ignores related to typing (#26254)

2025-10-05 16:37:55 +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

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

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

test_fusion_attn.py

[ROCm] Split AITER unified attention into its own backend (#25507)

2025-10-06 22:49:23 +00:00

test_fusion.py

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

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

test_noop_elimination.py

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

2025-10-05 07:06:22 -07: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

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

2025-10-06 05:12:40 +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

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

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