Amir Samani 8c1c81a3de

[core] add nccl symmetric memory for all reduce (#24532) ...

Signed-off-by: Amir Samani <asamani@nvidia.com>
Signed-off-by: Michael Goin <mgoin64@gmail.com>
Co-authored-by: Michael Goin <mgoin64@gmail.com>

2025-09-23 14:33:06 -04:00

..

__init__.py

[torch.compile] avoid Dynamo guard evaluation overhead (#7898)

2024-08-28 16:10:12 -07:00

activation_quant_fusion.py

[torch.compile] Cleanup compilation tests and custom passes, add debug utils, fix DCE bug (#23091), fix test (#24376), and prep for custom op matching (#24604) (#24542)

2025-09-22 12:30:05 -07:00

backends.py

[torch.compile] CUDAGraph Inductor partition integration (#24281)

2025-09-20 01:02:15 +00:00

base_static_graph.py

[Docs] Fix griffe warning in base_static_graph.py (#25018)

2025-09-17 08:49:19 +00:00

collective_fusion.py

[torch.compile] Cleanup compilation tests and custom passes, add debug utils, fix DCE bug (#23091), fix test (#24376), and prep for custom op matching (#24604) (#24542)

2025-09-22 12:30:05 -07:00

compiler_interface.py

Add test case for compiling multiple graphs (#21044)

2025-07-23 11:00:47 -07:00

counter.py

[Bugfix] VLLM_V1 supports passing other compilation levels (#19340)

2025-07-29 09:35:58 -04:00

cuda_graph.py

[core] add nccl symmetric memory for all reduce (#24532)

2025-09-23 14:33:06 -04:00

cuda_piecewise_backend.py

[Core] Allow full cudagraph with separate attention routines and orthogonal to compilation, add support for FA2 and FlashInfer (#20059)

2025-08-15 10:01:39 -04:00

decorators.py

[torch.compile] CUDAGraph Inductor partition integration (#24281)

2025-09-20 01:02:15 +00:00

fix_functionalization.py

[torch.compile] Cleanup compilation tests and custom passes, add debug utils, fix DCE bug (#23091), fix test (#24376), and prep for custom op matching (#24604) (#24542)

2025-09-22 12:30:05 -07:00

fusion_attn.py

[torch.compile] Cleanup compilation tests and custom passes, add debug utils, fix DCE bug (#23091), fix test (#24376), and prep for custom op matching (#24604) (#24542)

2025-09-22 12:30:05 -07:00

fusion.py

[torch.compile] Cleanup compilation tests and custom passes, add debug utils, fix DCE bug (#23091), fix test (#24376), and prep for custom op matching (#24604) (#24542)

2025-09-22 12:30:05 -07:00

fx_utils.py

[torch.compile][ROCm] Fuse quantization onto attention using a torch.compile pass (#16756)

2025-06-12 08:31:04 -07:00

inductor_pass.py

[Bugfix] Add fake mode around passes (#23349)

2025-08-28 11:25:56 -04:00

monitor.py

[Doc]: fix typos in Python comments (#24001)

2025-08-31 08:21:59 +00:00

noop_elimination.py

[torch.compile] Cleanup compilation tests and custom passes, add debug utils, fix DCE bug (#23091), fix test (#24376), and prep for custom op matching (#24604) (#24542)

2025-09-22 12:30:05 -07:00

pass_manager.py

[torch.compile] Cleanup compilation tests and custom passes, add debug utils, fix DCE bug (#23091), fix test (#24376), and prep for custom op matching (#24604) (#24542)

2025-09-22 12:30:05 -07:00

post_cleanup.py

[torch.compile] Cleanup compilation tests and custom passes, add debug utils, fix DCE bug (#23091), fix test (#24376), and prep for custom op matching (#24604) (#24542)

2025-09-22 12:30:05 -07:00

sequence_parallelism.py

[torch.compile] Cleanup compilation tests and custom passes, add debug utils, fix DCE bug (#23091), fix test (#24376), and prep for custom op matching (#24604) (#24542)

2025-09-22 12:30:05 -07:00

torch25_custom_graph_pass.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

vllm_inductor_pass.py

[torch.compile] Cleanup compilation tests and custom passes, add debug utils, fix DCE bug (#23091), fix test (#24376), and prep for custom op matching (#24604) (#24542)

2025-09-22 12:30:05 -07:00

wrapper.py

[Bugfix] Remove VLLM_TEST_DYNAMO_FULLGRAPH_CAPTURE #2969 (#25090)

2025-09-19 20:27:21 -04:00