Valentine233 02c7ab2f9b [cpp wrapper] add AOTI shim for collective ops (#154492) ...

Implementations:
1. Move collective ops to c10d namespace, so that we can call them externally.
2. Add AOTI shims for collective ops.

Testing
1. Add c10d functional UT for cpu.
2. Include the above one in cpp wrapper UT.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/154492
Approved by: https://github.com/desertfire

2025-06-25 01:20:05 +00:00

..

autoheuristic

[BE][Ez]: Optimize unnecessary lambda with operator (#154722)

2025-05-30 23:47:10 +00:00

codegen

[Inductor] Restrict block analysis to only match integer dims and strides (#149615)

2025-06-24 22:43:12 +00:00

compile_worker

Add logging for async compile worker statistics (#155820)

2025-06-19 23:10:15 +00:00

fx_passes

Revert "[invoke_subgraph] make same subgraph share get_attr target (#156260)"

2025-06-23 18:24:10 +00:00

kernel

Unify dynamic shapes APIs naming 2 (expect_true and check) attempt2 (#156518)

2025-06-24 21:01:38 +00:00

package

[aoti] package weights to disk and dedup (#155241)

2025-06-19 17:17:17 +00:00

runtime

Revert "[logging] dynamo_timed for CachingAutotuner.coordinate_descent_tuning (#156517)"

2025-06-24 23:45:13 +00:00

__autotune_main__.py

Improve subproc autotuning implementation (#149700)

2025-03-28 01:06:39 +00:00

__init__.py

[aoti] package weights to disk and dedup (#155241)

2025-06-19 17:17:17 +00:00

analyze_preserves_zero_mask.py

Revert two recent prologue prs (#151013)

2025-04-10 23:48:41 +00:00

aoti_eager.py

PEP585 update - torch/_inductor/[_-i]* (#145137)

2025-01-19 01:22:47 +00:00

async_compile.py

Add logging for async compile worker statistics (#155820)

2025-06-19 23:10:15 +00:00

autotune_process.py

[BE][3/16] fix typos in torch/ (torch/_inductor/) (#156313)

2025-06-23 02:57:12 +00:00

bounds.py

[inductor] Refactor op handlers part 5 (#146257)

2025-02-08 18:00:30 +00:00

choices.py

FlexAttn config refactor + ROCm optimisations (#156307)

2025-06-22 22:27:38 +00:00

codecache.py

[BE][3/16] fix typos in torch/ (torch/_inductor/) (#156313)

2025-06-23 02:57:12 +00:00

comm_analysis.py

[cpp wrapper] add AOTI shim for collective ops (#154492)

2025-06-25 01:20:05 +00:00

comm_lowering.py

[cpp wrapper] add AOTI shim for collective ops (#154492)

2025-06-25 01:20:05 +00:00

comms.py

[BE][3/16] fix typos in torch/ (torch/_inductor/) (#156313)

2025-06-23 02:57:12 +00:00

compile_fx_async.py

Use correct boxed_forward_device_index when running CompiledFxGraph.post_compile (#148130)

2025-03-23 02:57:58 +00:00

compile_fx_ext.py

Migrate from lru_cache to cache (#155613)

2025-06-11 19:44:18 +00:00

compile_fx_subproc.py

Rename inductor cache (#156128)

2025-06-17 03:57:18 +00:00

compile_fx.py

[aoti] package weights to disk and dedup (#155241)

2025-06-19 17:17:17 +00:00

compiler_bisector.py

Migrate from lru_cache to cache (#155613)

2025-06-11 19:44:18 +00:00

config.py

[CUDAGraph] add config cudagraph_capture_sizes (#156551)

2025-06-24 05:14:49 +00:00

constant_folding.py

Add dont constant fold flag (#154945)

2025-06-10 14:52:26 +00:00

cpp_builder.py

[BE][3/16] fix typos in torch/ (torch/_inductor/) (#156313)

2025-06-23 02:57:12 +00:00

cpu_vec_isa.py

mypy 1.16.0 (#155821)

2025-06-14 18:18:43 +00:00

cudagraph_trees.py

[CUDAGraph] add config cudagraph_capture_sizes (#156551)

2025-06-24 05:14:49 +00:00

cudagraph_utils.py

[BE][3/16] fix typos in torch/ (torch/_inductor/) (#156313)

2025-06-23 02:57:12 +00:00

custom_graph_pass.py

[BE][3/16] fix typos in torch/ (torch/_inductor/) (#156313)

2025-06-23 02:57:12 +00:00

debug.py

[inductor][cutlass backend] Log prescreening elpase (#155508)

2025-06-12 16:48:52 +00:00

decomposition.py

Improve torch.ops typing (#154555)

2025-06-22 15:52:27 +00:00

dependencies.py

[BE][3/16] fix typos in torch/ (torch/_inductor/) (#156313)

2025-06-23 02:57:12 +00:00

dtype_propagation.py

Migrate from lru_cache to cache (#155613)

2025-06-11 19:44:18 +00:00

exc.py

PEP585 update - torch/_inductor/[_-i]* (#145137)

2025-01-19 01:22:47 +00:00

extern_node_serializer.py

[BE][AOTI] Remove duplicate schema for ExternKernelNode (#155867)

2025-06-14 02:03:27 +00:00

freezing_utils.py

PEP585: More UP006 fixes (#146392)

2025-02-20 06:18:13 +00:00

freezing.py

mypy 1.16.0 (#155821)

2025-06-14 18:18:43 +00:00

fuzzer.py

[AOTI][reland] Add an option to specify custom op C shim (#153968)

2025-05-21 15:57:57 +00:00

fx_utils.py

Revert "Inductor logging + analysis of torch.profile (#149697)"

2025-06-10 15:38:40 +00:00

graph.py

[BE][3/16] fix typos in torch/ (torch/_inductor/) (#156313)

2025-06-23 02:57:12 +00:00

hooks.py

PEP585 update - torch/_inductor/[_-i]* (#145137)

2025-01-19 01:22:47 +00:00

index_propagation.py

[BE][3/16] fix typos in torch/ (torch/_inductor/) (#156313)

2025-06-23 02:57:12 +00:00

inductor_prims.py

[inductor] lowering for fractional_max_pool3d (#148630)

2025-05-22 16:06:29 +00:00

ir.py

[cpp wrapper] add AOTI shim for collective ops (#154492)

2025-06-25 01:20:05 +00:00

jagged_lowerings.py

Revert "[inductor] Add typing to _inductor/ir.py (#149958)"

2025-06-06 15:19:16 +00:00

loop_body.py

Migrate from lru_cache to cache (#155613)

2025-06-11 19:44:18 +00:00

lowering.py

Unify dynamic shapes APIs naming 2 (expect_true and check) attempt2 (#156518)

2025-06-24 21:01:38 +00:00

memory.py

[PT2][memory] correct wait tensor output size (#153569)

2025-06-04 17:49:25 +00:00

metrics.py

Replace runtime type parameterization (#155221)

2025-06-05 21:43:54 +00:00

mkldnn_ir.py

[BE][3/16] fix typos in torch/ (torch/_inductor/) (#156313)

2025-06-23 02:57:12 +00:00

mkldnn_lowerings.py

[BE][3/16] fix typos in torch/ (torch/_inductor/) (#156313)

2025-06-23 02:57:12 +00:00

mock_cache.py

PEP585 update - torch/_inductor (#145198)

2025-01-21 21:04:33 +00:00

ops_handler.py

Revert "[inductor] Add typing to _inductor/ir.py (#149958)"

2025-06-06 15:19:16 +00:00

optimize_indexing.py

PEP585 update - torch/_inductor (#145198)

2025-01-21 21:04:33 +00:00

output_code.py

[aoti] package weights to disk and dedup (#155241)

2025-06-19 17:17:17 +00:00

pattern_matcher.py

Revert "[invoke_subgraph] make same subgraph share get_attr target (#156260)"

2025-06-23 18:24:10 +00:00

quantized_lowerings.py

[BE][1/X] Phase out usage of use_max_autotune() (#155847)

2025-06-14 03:16:20 +00:00

remote_cache.py

[BE][3/16] fix typos in torch/ (torch/_inductor/) (#156313)

2025-06-23 02:57:12 +00:00

scheduler.py

[Inductor] Fix epilogue fusion decision with 1 Triton caller as choice (#156500)

2025-06-24 22:33:47 +00:00

script.ld

Place .lrodata later in the binary (#117575)

2024-01-18 17:58:18 +00:00

select_algorithm.py

[ez] fix typo in select_algorithm.py (#156625)

2025-06-24 23:01:58 +00:00

sizevars.py

Unify dynamic shapes APIs naming 2 (expect_true and check) attempt2 (#156518)

2025-06-24 21:01:38 +00:00

standalone_compile.py

[BE][3/16] fix typos in torch/ (torch/_inductor/) (#156313)

2025-06-23 02:57:12 +00:00

subgraph_lowering.py

[inductor] Refactor op handlers part 5 (#146257)

2025-02-08 18:00:30 +00:00

template_heuristics.py

[Inductor] Allow exhaustive autotuning across all GEMM options (#156610)

2025-06-24 01:42:05 +00:00

test_case.py

Rename inductor cache (#156128)

2025-06-17 03:57:18 +00:00

test_operators.py

[CI] Fix GPUTests.test_scheduler_vertical_fusion1 (#151166)

2025-04-13 00:41:51 +00:00

tiling_utils.py

[BE][3/16] fix typos in torch/ (torch/_inductor/) (#156313)

2025-06-23 02:57:12 +00:00

triton_bundler.py

[BE][3/16] fix typos in torch/ (torch/_inductor/) (#156313)

2025-06-23 02:57:12 +00:00

utils.py

[cpp wrapper] add AOTI shim for collective ops (#154492)

2025-06-25 01:20:05 +00:00

virtualized.py

[inductor] Add a helper for convert index_dtype to torch dtype (#149531)

2025-03-20 21:33:29 +00:00

wrapper_benchmark.py

Revert "Inductor logging + analysis of torch.profile (#149697)"

2025-06-10 15:38:40 +00:00