pytorch

autoheuristic

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

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

codegen

Fix broken URLs (#152237 )

2025-04-27 09:56:42 +00:00

compile_worker

Cache the value of torch_key in subproc (#151057 )

2025-04-22 18:54:06 +00:00

fx_passes

Fix broken URLs (#152237 )

2025-04-27 09:56:42 +00:00

kernel

Fix broken URLs (#152237 )

2025-04-27 09:56:42 +00:00

package

[AOTInductor] Add interface for user managed buffer in package api. (#151325 )

2025-04-16 04:25:40 +00:00

runtime

Fix broken URLs (#152237 )

2025-04-27 09:56:42 +00:00

__autotune_main__.py

Improve subproc autotuning implementation (#149700 )

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

__init__.py

[standalone_compile] Dynamic shape handling (#151788 )

2025-04-22 20:17:24 +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

Turn on static cuda launcher in OSS (#151691 )

2025-04-25 17:48:53 +00:00

autotune_process.py

[pytorch][triton] Warp specialization support in TritonTemplate for torchinductor (#148503 ) (#150122 )

2025-03-29 03:36:50 +00:00

bounds.py

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

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

choices.py

Reland "Introduce new template heuristic for triton autotune configs" (#147452 )

2025-03-26 15:47:06 +00:00

codecache.py

Better error msg for too big to optimize (#151855 )

2025-04-24 16:35:19 +00:00

comm_analysis.py

[BE][Easy][16/19] enforce style for empty lines in import segments in torch/_i*/ (#129768 )

2024-07-20 16:20:58 +00:00

comm_lowering.py

Fix an issue where functional collectives don't force fx stride on inputs when compiled (#146467 )

2025-02-10 19:15:49 +00:00

comms.py

[inductor][comms] fix node_summary for composite scheduler nodes (#150258 )

2025-04-16 10:18:33 +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

async fx compile (#146135 )

2025-03-19 14:07:51 +00:00

compile_fx_subproc.py

async fx compile (#146135 )

2025-03-19 14:07:51 +00:00

compile_fx.py

Package const folded graph's cubin file (#152145 )

2025-04-25 18:38:32 +00:00

compiler_bisector.py

Add a couple config options to compiler bisector (#148450 )

2025-03-04 23:23:21 +00:00

config.py

Fix broken URLs (#152237 )

2025-04-27 09:56:42 +00:00

constant_folding.py

skip torchbind in cosntant folding (#148993 )

2025-03-12 18:08:08 +00:00

cpp_builder.py

cpp_wrapper: Miscellaneous fixups (#150143 )

2025-04-10 03:31:12 +00:00

cpu_vec_isa.py

Revert "Extend vec backend with BF16 SVE intrinsics (#143666 )"

2025-03-24 18:13:50 +00:00

cudagraph_trees.py

[logging] Clean up dynamo_timed usages in cudagraph_trees (#152136 )

2025-04-25 19:18:12 +00:00

cudagraph_utils.py

[CUDAGraph] support meta tensor (#150478 )

2025-04-02 07:21:50 +00:00

custom_graph_pass.py

Migrate from Tuple -> tuple in torch/_inductor (#144264 )

2025-01-07 03:27:27 +00:00

debug.py

Fix only logging ir_post_fusion with torch_compile_debug enabled (#148499 )

2025-03-05 05:35:09 +00:00

decomposition.py

[MPSInductor] Disable mm/bmm decompositions (#150541 )

2025-04-02 16:07:18 +00:00

dependencies.py

[Graph Partition] Support symbol inputs (#149458 )

2025-03-26 17:21:30 +00:00

dtype_propagation.py

Remove libdevice ops in inductor (#151562 )

2025-04-17 22:18:00 +00:00

exc.py

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

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

extern_node_serializer.py

Back out "[AOTI] Always use oss schema for ExternKernelNodes serialization" (#151026 )

2025-04-10 22:36:35 +00:00

freezing_utils.py

PEP585: More UP006 fixes (#146392 )

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

freezing.py

Fix code cache + freezing compile-time regression (#145868 )

2025-01-31 02:04:15 +00:00

fuzzer.py

[dynamo] Always trace into tensor subclass __torch_function__ (#149792 )

2025-04-02 20:57:00 +00:00

fx_utils.py

[BE][PYFMT] migrate PYFMT for torch._inductor to ruff format (#144550 )

2025-02-28 13:33:19 +00:00

graph.py

Fix Issues in deferring runtime assertions. (#151170 )

2025-04-16 08:10:17 +00:00

hooks.py

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

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

index_propagation.py

[BE][PYFMT] migrate PYFMT for torch._inductor to ruff format (#144550 )

2025-02-28 13:33:19 +00:00

inductor_prims.py

[inductor] Lowerings for max_pool3d (#148210 )

2025-04-02 14:13:01 +00:00

ir.py

[inductor] handle offset in ReinterpretView for alignment (#151859 )

2025-04-23 01:50:49 +00:00

jagged_lowerings.py

PEP585 update - torch/_inductor (#145198 )

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

loop_body.py

[ez] fix typo in comment (#151755 )

2025-04-21 14:52:39 +00:00

lowering.py

Fix broken URLs (#152237 )

2025-04-27 09:56:42 +00:00

memory.py

[graph partition] support graphsafe_run_with_rng_state (#150958 )

2025-04-12 03:17:08 +00:00

metrics.py

[Inductor] Support parallel reduction for GroupNorm (#144020 )

2025-03-01 17:11:50 +00:00

mkldnn_ir.py

[Quant][PT2E][X86] enable qconv1d-relu fusion (#150751 )

2025-04-09 14:42:02 +00:00

mkldnn_lowerings.py

Fix broken URLs (#152237 )

2025-04-27 09:56:42 +00:00

mock_cache.py

PEP585 update - torch/_inductor (#145198 )

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

ops_handler.py

Remove libdevice ops in inductor (#151562 )

2025-04-17 22:18:00 +00:00

optimize_indexing.py

PEP585 update - torch/_inductor (#145198 )

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

output_code.py

Unpack the output code in the standalone_compile (#151609 )

2025-04-21 17:37:38 +00:00

pattern_matcher.py

Rename node.meta["arg_kwarg_vals"] to node.meta["eager_input_vals"] (#148092 )

2025-04-02 13:18:04 +00:00

quantized_lowerings.py

Add AOTI shim for _weight_int4pack_mm_cpu_tensor (#149031 )

2025-03-18 01:33:13 +00:00

remote_cache.py

[Indcutor Remote Cache] Raise an exception if redis module is required but not available (#151779 )

2025-04-26 11:21:54 +00:00

scheduler.py

Fix broken URLs (#152237 )

2025-04-27 09:56:42 +00:00

script.ld

Place .lrodata later in the binary (#117575 )

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

select_algorithm.py

Revert two recent prologue prs (#151013 )

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

sizevars.py

[BE][PYFMT] migrate PYFMT for torch._inductor to ruff format (#144550 )

2025-02-28 13:33:19 +00:00

standalone_compile.py

Check integrity of bytes in AppendingByteSerializer (#152139 )

2025-04-26 18:10:58 +00:00

subgraph_lowering.py

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

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

template_heuristics.py

[Inductor] Add Additional Configs for persistent+TMA version of Triton mm and addmm (#150587 )

2025-04-23 18:21:35 +00:00

test_case.py

[Inductor] be able to disable cache for test (#141195 )

2025-01-24 19:15:55 +00:00

test_operators.py

[CI] Fix GPUTests.test_scheduler_vertical_fusion1 (#151166 )

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

triton_bundler.py

hack to try to fix not empty triton dir (#151119 )

2025-04-12 03:21:41 +00:00

utils.py

Fix broken URLs (#152237 )

2025-04-27 09:56:42 +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

[BE][PYFMT] migrate PYFMT for torch._inductor to ruff format (#144550 )

2025-02-28 13:33:19 +00:00