frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity
Files
2b71b62045fdcd89bcae050cd7bacef39988f8c1
pytorch/torch/_inductor
History
eellison 2b71b62045 Add Memory Estimation Tracker (#165059) 
Add Memory Tracker utility, which will track live memory given alternate ordering of nodes.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/165059
Approved by: https://github.com/ezyang, https://github.com/IvanKobzarev
ghstack dependencies: #164738, #164783, #164944, #164945
2025-10-15 19:44:29 +00:00
..
analysis
Add suppressions to torch/_inductor (#165062)
2025-10-09 20:34:20 +00:00
autoheuristic
Update ruff to 0.13.1 (#163744)
2025-09-26 10:12:21 +00:00
codegen
Revert "[inductor] Expand use of generic benchmark function (#164938)"
2025-10-15 17:48:38 +00:00
compile_worker
Add suppressions to torch/_inductor (#165062)
2025-10-09 20:34:20 +00:00
fx_passes
Add Memory Estimation Tracker (#165059)
2025-10-15 19:44:29 +00:00
kernel
Revert "[Inductor][CuTeDSL] Move load_template up two directories (#165347)"
2025-10-15 19:30:46 +00:00
package
Add suppressions to torch/_inductor (#165062)
2025-10-09 20:34:20 +00:00
runtime
Revert "[inductor] Expand use of generic benchmark function (#164938)"
2025-10-15 17:48:38 +00:00
template_heuristics
Add suppressions to torch/_inductor (#165062)
2025-10-09 20:34:20 +00:00
__autotune_main__.py
Improve subproc autotuning implementation (#149700)
2025-03-28 01:06:39 +00:00
__init__.py
Add suppressions to torch/_inductor (#165062)
2025-10-09 20:34:20 +00:00
analyze_preserves_zero_mask.py
Add suppressions to torch/_inductor (#165062)
2025-10-09 20:34:20 +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
Limit coll bucketing within node idxs (#164944)
2025-10-08 22:18:53 +00:00
autotune_process.py
Add suppressions to torch/_inductor (#165062)
2025-10-09 20:34:20 +00:00
await_utils.py
Add suppressions to torch/_inductor (#165062)
2025-10-09 20:34:20 +00:00
bounds.py
[inductor] Refactor op handlers part 5 (#146257)
2025-02-08 18:00:30 +00:00
cache.py
Add suppressions to torch/_inductor (#165062)
2025-10-09 20:34:20 +00:00
choices.py
Add suppressions to torch/_inductor (#165062)
2025-10-09 20:34:20 +00:00
codecache.py
Make LOCK_TIMEOUT in codecache configurable (#165030)
2025-10-10 20:22:11 +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
Add suppressions to torch/_inductor (#165062)
2025-10-09 20:34:20 +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
Add suppressions to torch/_inductor (#165062)
2025-10-09 20:34:20 +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
Add suppressions to torch/_inductor (#165062)
2025-10-09 20:34:20 +00:00
compile_fx_subproc.py
Rename inductor cache (#156128)
2025-06-17 03:57:18 +00:00
compile_fx.py
[inductor] verify determinism with inductor benchmark script (#164904)
2025-10-12 00:03:42 +00:00
compiler_bisector.py
Add suppressions to torch/_inductor (#165062)
2025-10-09 20:34:20 +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
[2/N] More ruff SIM fixes (#165031)
2025-10-14 14:22:54 +00:00
constant_folding.py
[2/N] Use "is" in python type comparison (#165142)
2025-10-10 15:36:44 +00:00
cpp_builder.py
Add an option to put store large mmap weights on disk (#164526)
2025-10-10 07:53:57 +00:00
cpu_vec_isa.py
Revert "Add SVE128 ISA (#158932)"
2025-10-10 01:17:02 +00:00
cudagraph_trees.py
Add suppressions to torch/_inductor (#165062)
2025-10-09 20:34:20 +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
[inductor] custom_graph_pass.get_hash_for_files: don't hash paths (#165020)
2025-10-09 20:07:53 +00:00
debug.py
Add suppressions to torch/_inductor (#165062)
2025-10-09 20:34:20 +00:00
decomposition.py
[Inductor] Fix out-of-bounds indices in repeat_interleave decomposition (#165368)
2025-10-14 21:16:36 +00:00
dependencies.py
Add suppressions to torch/_inductor (#165062)
2025-10-09 20:34:20 +00:00
dtype_propagation.py
Native matmul (#157743)
2025-10-14 04:22:30 +00:00
exc.py
Add suppressions to torch/_inductor (#165062)
2025-10-09 20:34:20 +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
Add suppressions to torch/_inductor (#165062)
2025-10-09 20:34:20 +00:00
fuzzer.py
Enable ruff rule E721 (#165162)
2025-10-13 01:48:55 +00:00
fx_utils.py
[1/N] Use "is" in python type comparison (#165037)
2025-10-10 12:36:50 +00:00
graph.py
[Bugfix][Inductor][Dynamo] Fix stride incorrectness issues for stride 0 tensor (#164897)
2025-10-10 21:26:57 +00:00
hooks.py
PEP585 update - torch/_inductor/[_-i]* (#145137)
2025-01-19 01:22:47 +00:00
index_propagation.py
Add suppressions to torch/_inductor (#165062)
2025-10-09 20:34:20 +00:00
inductor_prims.py
[Easy] Add pointwise tag to fma (#164149)
2025-09-29 22:40:04 +00:00
ir.py
Revert "[inductor] Expand use of generic benchmark function (#164938)"
2025-10-15 17:48:38 +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
Add suppressions to torch/_inductor (#165062)
2025-10-09 20:34:20 +00:00
lowering.py
Add suppressions to torch/_inductor (#165062)
2025-10-09 20:34:20 +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
[inductor] log kernel autotuning result to a csv (#164191)
2025-10-02 00:25:34 +00:00
mkldnn_ir.py
Add suppressions to torch/_inductor (#165062)
2025-10-09 20:34:20 +00:00
mkldnn_lowerings.py
[2/N] Use "is" in python type comparison (#165142)
2025-10-10 15:36:44 +00:00
mock_cache.py
PEP585 update - torch/_inductor (#145198)
2025-01-21 21:04:33 +00:00
ops_handler.py
Native matmul (#157743)
2025-10-14 04:22:30 +00:00
optimize_indexing.py
PEP585 update - torch/_inductor (#145198)
2025-01-21 21:04:33 +00:00
output_code.py
Add suppressions to torch/_inductor (#165062)
2025-10-09 20:34:20 +00:00
pattern_matcher.py
Add suppressions to torch/_inductor (#165062)
2025-10-09 20:34:20 +00:00
quantized_lowerings.py
Add suppressions to torch/_inductor (#165062)
2025-10-09 20:34:20 +00:00
remote_cache.py
Add suppressions to torch/_inductor (#165062)
2025-10-09 20:34:20 +00:00
remote_gemm_autotune_cache.py
integrate kernacle into inductor (#160121)
2025-08-08 02:14:44 +00:00
scheduler.py
Revert "[inductor] Expand use of generic benchmark function (#164938)"
2025-10-15 17:48:38 +00:00
script.ld
Place .lrodata later in the binary (#117575)
2024-01-18 17:58:18 +00:00
select_algorithm.py
Revert "[inductor] Expand use of generic benchmark function (#164938)"
2025-10-15 17:48:38 +00:00
shape_propagation.py
Native matmul (#157743)
2025-10-14 04:22:30 +00:00
sizevars.py
[inductor] fix issue for example value with unbacked strides (#163660)
2025-10-14 20:07:51 +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
Add suppressions to torch/_inductor (#165062)
2025-10-09 20:34:20 +00:00
tiling_utils.py
Add suppressions to torch/_inductor (#165062)
2025-10-09 20:34:20 +00:00
triton_bundler.py
[inductor][triton] more JITCallable._hash_lock support (#162244)
2025-09-08 07:57:48 +00:00
utils.py
Revert "[Inductor][CuTeDSL] Move load_template up two directories (#165347)"
2025-10-15 19:30:46 +00:00
virtualized.py
[export] Fix custom ops in subgraphs (#160004)
2025-08-18 15:42:19 +00:00
wrapper_benchmark.py
Revert "[inductor] Expand use of generic benchmark function (#164938)"
2025-10-15 17:48:38 +00:00