pytorch

mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00

Files

Yavuz Yetim 7afcb030d8 Back out "Revert D81959389" (#163905 )

Summary:
Original commit changeset: 06888d7ebff0

Original Phabricator Diff: D82932788

Restricted the test to SM90 for scaled_grouped_mm

Test Plan: TBD (will share the linux CI results)

Differential Revision: D83283991

Pull Request resolved: https://github.com/pytorch/pytorch/pull/163905
Approved by: https://github.com/angelayi

2025-09-30 07:05:13 +00:00

analysis

Revert "Improve device info with new flops and bandwidth formula based on hardware libraries (#162245 )"

2025-09-19 20:09:12 +00:00

autoheuristic

Update ruff to 0.13.1 (#163744 )

2025-09-26 10:12:21 +00:00

codegen

Remove unnecessary list comprehensions (#164103 )

2025-09-30 03:56:54 +00:00

compile_worker

Allow for using a dedicated binary for the torch subproc pool. (#162093 )

2025-09-05 01:43:46 +00:00

fx_passes

Add Comm-Compute Preserving Bucketer (#163960 )

2025-09-30 04:53:58 +00:00

kernel

Back out "Revert D81959389" (#163905 )

2025-09-30 07:05:13 +00:00

package

Grab bag of (mostly) typing improvements (#158075 )

2025-07-21 19:17:01 +00:00

runtime

Revert "Enable outer reductions in fbcode (#163884 )"

2025-09-30 03:42:24 +00:00

template_heuristics

[Max Autotune][B200] Add addmm config to avoid test OOM (#164020 )

2025-09-29 17:38:46 +00:00

__autotune_main__.py

Improve subproc autotuning implementation (#149700 )

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

__init__.py

shape guards (#161178 )

2025-09-08 22:44:09 +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

[multi-kernel] shape-similarity kernel selection (#163090 )

2025-09-23 21:00:47 +00:00

augmented_graph_helper.py

Helper to augment graph with additional deps (#163959 )

2025-09-30 04:53:58 +00:00

autotune_process.py

Add cutedsl template support to compile (#160108 )

2025-08-18 04:37:15 +00:00

await_utils.py

integrate kernacle into inductor (#160121 )

2025-08-08 02:14:44 +00:00

bounds.py

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

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

cache.py

[pt2][cache] rework cache for true generic usage + better tests (#163488 )

2025-09-23 07:31:48 +00:00

choices.py

Less aggressive persistent reduction when it could induce large masking with dynamic shapes (#163365 )

2025-09-23 21:58:57 +00:00

codecache.py

[aoti] Save compute information (#163792 )

2025-09-26 05:40:44 +00:00

comm_analysis.py

[inductor] do comm compute overlap at aten fx level (#163215 )

2025-09-30 04:53:58 +00:00

comm_lowering.py

[AOTI] Fix memory leak from all_reduce (#159818 )

2025-08-06 18:11:14 +00:00

comms_debug.py

[inductor] Estimate peak memory allocfree and applying to reordering collectives (#160113 )

2025-08-22 14:19:57 +00:00

comms.py

[inductor] Runtime estimations: use nccl estimator; mm only benchmark mode (#161405 )

2025-09-08 14:33:19 +00:00

compile_fx_async.py

[pc] introduce ProgressiveCompilationState and clear callback (#157619 )

2025-07-05 07:55:11 +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

[BE] Delete all pre py-3.10 checks (#163653 )

2025-09-23 23:22:53 +00:00

compiler_bisector.py

Migrate from lru_cache to cache (#155613 )

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

config_comms.py

[inductor] Runtime estimations: use nccl estimator; mm only benchmark mode (#161405 )

2025-09-08 14:33:19 +00:00

config.py

Add Comm-Compute Preserving Bucketer (#163960 )

2025-09-30 04:53:58 +00:00

constant_folding.py

Add dont constant fold flag (#154945 )

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

cpp_builder.py

Fix windows path escape characters (#162761 )

2025-09-17 23:39:39 +00:00

cpu_vec_isa.py

Add SVE128 ISA (#158932 )

2025-09-29 14:49:19 +00:00

cudagraph_trees.py

Fix types in graphs.py (#158192 )

2025-07-15 19:49:38 +00:00

cudagraph_utils.py

[CUDAGraph] add config to error on skipping cudagraph (#161862 )

2025-09-04 15:52:39 +00:00

custom_graph_pass.py

allow user to pass in custom partitioner function (#157580 )

2025-09-05 22:49:39 +00:00

debug.py

Refactor Provenance Tracking (#163378 )

2025-09-25 22:55:59 +00:00

decomposition.py

[Inductor] address comments from https://github.com/pytorch/pytorch/pull/163803 (#163901 )

2025-09-26 17:18:44 +00:00

dependencies.py

[Inductor] don't call sympy_str when not needed (#162126 )

2025-09-19 20:21:33 +00:00

dtype_propagation.py

[Inductor] Update DeviceAssert op to behave like store (#163696 )

2025-09-24 23:35:56 +00:00

exc.py

[BE] Save attributes for CppCompileError for pickleing (#160294 )

2025-08-12 22:03:36 +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

[BE] Delete all pre py-3.10 checks (#163653 )

2025-09-23 23:22:53 +00:00

fx_utils.py

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

2025-07-07 22:13:34 +00:00

graph.py

[inductor] skip bmm when converting channel last (#159459 )

2025-09-26 09:11:40 +00:00

hooks.py

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

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

index_propagation.py

[inductor] Follow integer overflow rules in TypedExpr (#161922 )

2025-09-03 18:33:18 +00:00

inductor_prims.py

[Easy] Add pointwise tag to fma (#164149 )

2025-09-29 22:40:04 +00:00

ir.py

[inductor] do comm compute overlap at aten fx level (#163215 )

2025-09-30 04:53:58 +00:00

jagged_lowerings.py

Fix XPU CI UT test_circular_dependencies (#158189 )

2025-07-13 09:30:57 +00:00

kernel_inputs.py

[inductor][template heuristics] don't take layout to generate choices (#162238 )

2025-09-09 17:17:04 +00:00

kernel_template_choice.py

[inductor][choices] move extra kwargs out of get_template_configs (#163209 )

2025-09-20 05:30:40 +00:00

loop_body.py

[inductor] avoid creating LoopBody twice (#162101 )

2025-09-19 20:21:33 +00:00

lowering.py

[dynamic shapes] unbacked-safe slicing (#161414 )

2025-09-30 01:15:19 +00:00

memory.py

An improved heuristic for operator reordering for peak memory + debugging logs (#161810 )

2025-09-13 00:42:32 +00:00

metrics.py

Replace runtime type parameterization (#155221 )

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

mkldnn_ir.py

[AOTI][Intel GPU] Add XPU quantization ops to AOT Inductor. (#156572 )

2025-08-29 09:19:44 +00:00

mkldnn_lowerings.py

[Inductor][Float8] Add float8_e4m3fn into assertion dtype list. (#157684 )

2025-07-15 06:02:01 +00:00

mock_cache.py

PEP585 update - torch/_inductor (#145198 )

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

ops_handler.py

[Inductor] Update DeviceAssert op to behave like store (#163696 )

2025-09-24 23:35:56 +00:00

optimize_indexing.py

PEP585 update - torch/_inductor (#145198 )

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

output_code.py

[Inductor] Check if profiling before using record_function in CompiledFxGraph (#163747 )

2025-09-26 06:49:40 +00:00

pattern_matcher.py

[inductor] add subsystem to pattern matcher (#163922 )

2025-09-28 03:15:23 +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

pt2_remote_cache: Log sample for failures, and log the explicit reason we're faling. (#156874 )

2025-07-18 20:28:27 +00:00

remote_gemm_autotune_cache.py

integrate kernacle into inductor (#160121 )

2025-08-08 02:14:44 +00:00

scheduler.py

Refactor Provenance Tracking (#163378 )

2025-09-25 22:55:59 +00:00

script.ld

Place .lrodata later in the binary (#117575 )

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

select_algorithm.py

Remove unnecessary list comprehensions (#164103 )

2025-09-30 03:56:54 +00:00

shape_propagation.py

[Inductor] Add DeviceAssert op to enable device-side assertion in torch.compile (#160677 )

2025-08-28 18:57:34 +00:00

sizevars.py

[inductor] Fix unbounded number of substitutions when equality checks contain Max expr (#163685 )

2025-09-26 18:46:36 +00:00

standalone_compile.py

[standalone_compile] binary format write should be atomic (#162432 )

2025-09-09 18:43:13 +00:00

subgraph_lowering.py

[FlexAttention] fixing learnable bias assertion error in inductor (#161170 )

2025-08-23 06:24:22 +00:00

test_case.py

Rename inductor cache (#156128 )

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

test_operators.py

[BE] remove torch deploy - conditionals (#158288 )

2025-07-29 17:40:49 +00:00

tiling_utils.py

[BE] Delete all pre py-3.10 checks (#163653 )

2025-09-23 23:22:53 +00:00

triton_bundler.py

[inductor][triton] more JITCallable._hash_lock support (#162244 )

2025-09-08 07:57:48 +00:00

utils.py

[inductor] pdl: enable launch and deduplicate waits (#162014 )

2025-09-29 16:10:26 +00:00

virtualized.py

[export] Fix custom ops in subgraphs (#160004 )

2025-08-18 15:42:19 +00:00

wrapper_benchmark.py

[inductor] Make times and repeat parameters command line args (#158590 )

2025-07-18 20:07:55 +00:00