Richard Zou 26b4fa45be

Add ability to use CUDAGraphs with use_inductor=False (#17345) ...

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

2025-05-29 10:16:52 +08:00

..

piecewise

Add ability to use CUDAGraphs with use_inductor=False (#17345)

2025-05-29 10:16:52 +08:00

__init__.py

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

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

backend.py

[Feature]Add async tensor parallelism using compilation pass (#17882)

2025-05-23 01:03:34 -07:00

conftest.py

[V1] V1 Enablement Oracle (#13726)

2025-03-14 22:02:20 -07:00

test_async_tp.py

[Feature]Add async tensor parallelism using compilation pass (#17882)

2025-05-23 01:03:34 -07:00

test_basic_correctness.py

[easy] Print number of needed GPUs in skip message (#17594)

2025-05-02 15:27:43 -07:00

test_full_graph.py

Improve configs - the rest! (#17562)

2025-05-09 15:18:44 -07:00

test_functionalization.py

Improve configs - the rest! (#17562)

2025-05-09 15:18:44 -07:00

test_fusion.py

[Feature]Add async tensor parallelism using compilation pass (#17882)

2025-05-23 01:03:34 -07:00

test_pass_manager.py

[Fix] check to make sure processor has chat templates (#18047)

2025-05-13 03:04:10 -07:00

test_sequence_parallelism.py

[Feature]Add async tensor parallelism using compilation pass (#17882)

2025-05-23 01:03:34 -07:00

test_silu_mul_quant_fusion.py

[AMD][torch.compile] Enable silu+fp8_quant fusion for rocm (#18082)

2025-05-13 22:13:56 -07:00

test_wrapper.py

[Misc] Add SPDX-License-Identifier headers to python source files (#12628)

2025-02-02 11:58:18 -08:00