OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 00:20:18 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
30587195d3
pytorch/torch/_inductor
History
Aaron Orenstein 250ae2531c Fix types in graphs.py (#158192) 
Added type annotations for torch/cuda/graphs.py

Pull Request resolved: https://github.com/pytorch/pytorch/pull/158192
Approved by: https://github.com/oulgen
2025-07-15 19:49:38 +00:00
..
analysis [BE][2/16] fix typos in torch/ (torch/_*/) (#156312) 2025-07-12 05:47:06 +00:00
autoheuristic [BE][Ez]: Optimize unnecessary lambda with operator (#154722) 2025-05-30 23:47:10 +00:00
codegen [AOTI][CPP] add flag TORCHINDUCTOR_CPP_FORCE_INLINE_KERNEL (#157949) 2025-07-15 10:51:43 +00:00
compile_worker [inductor] Quiesce Triton compile worker pool after each dynamo compile (#156187) 2025-07-08 22:53:13 +00:00
fx_passes [Optimus] add einsum_to_pointwise_pass pattern (#155666) 2025-07-15 17:50:23 +00:00
kernel Fix grouped MM output strides when compiled but not max-autotuned (#158143) 2025-07-15 11:53:13 +00:00
package [aoti] package weights to disk and dedup (#155241) 2025-06-19 17:17:17 +00:00
runtime [BE][2/16] fix typos in torch/ (torch/_*/) (#156312) 2025-07-12 05:47:06 +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
async_compile.py Add size_hints to cache key (#158026) 2025-07-11 01:47:50 +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
choices.py Revert "[PT2][fusion] ban fusions with large accumulated reads (#157563)" 2025-07-15 16:35:55 +00:00
codecache.py [cutlass backend] cache a few things for codegen and properties (#158158) 2025-07-15 00:18:31 +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 [inductor] Add typing to _inductor/ir.py (#149958) 2025-06-30 15:56:35 +00:00
comms.py Revert "[simple_fsdp][inductor_collectives] rewrite reorder_collectives, sink_waits_iterative (#158062)" 2025-07-15 16:31:13 +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 [AOTI] codegen for static linkage (#157129) 2025-07-10 16:03:50 +00:00
compiler_bisector.py Migrate from lru_cache to cache (#155613) 2025-06-11 19:44:18 +00:00
config.py Revert "[PT2][fusion] ban fusions with large accumulated reads (#157563)" 2025-07-15 16:35:55 +00:00
constant_folding.py Add dont constant fold flag (#154945) 2025-06-10 14:52:26 +00:00
cpp_builder.py [AOTI] codegen for static linkage (#157129) 2025-07-10 16:03:50 +00:00
cpu_vec_isa.py mypy 1.16.0 (#155821) 2025-06-14 18:18:43 +00:00
cudagraph_trees.py Fix types in graphs.py (#158192) 2025-07-15 19:49:38 +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 Revert "Fix full_like decomposition to preserve strides (#144765)" 2025-07-02 13:56:03 +00:00
dependencies.py Revert "[simple_fsdp][inductor_collectives] rewrite reorder_collectives, sink_waits_iterative (#158062)" 2025-07-15 16:31:13 +00:00
dtype_propagation.py Migrate from lru_cache to cache (#155613) 2025-06-11 19:44:18 +00:00
exc.py
extern_node_serializer.py [BE][AOTI] Remove duplicate schema for ExternKernelNode (#155867) 2025-06-14 02:03:27 +00:00
freezing_utils.py
freezing.py mypy 1.16.0 (#155821) 2025-06-14 18:18:43 +00:00
fuzzer.py Automatically load and save dynamo entries via caching_precompile (#155913) 2025-07-07 23:57:17 +00:00
fx_utils.py Inductor logging + analysis of torch.profile (#149697) 2025-07-07 22:13:34 +00:00
graph.py Revert "[PT2][fusion] ban fusions with large accumulated reads (#157563)" 2025-07-15 16:35:55 +00:00
hooks.py
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 Revert "[PT2][fusion] ban fusions with large accumulated reads (#157563)" 2025-07-15 16:35:55 +00:00
jagged_lowerings.py Fix XPU CI UT test_circular_dependencies (#158189) 2025-07-13 09:30:57 +00:00
loop_body.py Migrate from lru_cache to cache (#155613) 2025-06-11 19:44:18 +00:00
lowering.py [inductor] support unbacked symint in sdpfa (#157739) 2025-07-09 22:01:29 +00:00
memory.py Revert "[PT2][fusion] ban fusions with large accumulated reads (#157563)" 2025-07-15 16:35:55 +00:00
metrics.py Replace runtime type parameterization (#155221) 2025-06-05 21:43:54 +00:00
mkldnn_ir.py [inductor] Add typing to _inductor/ir.py (#149958) 2025-06-30 15:56:35 +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
ops_handler.py [inductor] Add typing to _inductor/ir.py (#149958) 2025-06-30 15:56:35 +00:00
optimize_indexing.py
output_code.py [aoti] package weights to disk and dedup (#155241) 2025-06-19 17:17:17 +00:00
pattern_matcher.py [Optimus] add einsum_to_pointwise_pass pattern (#155666) 2025-07-15 17:50: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 [BE][3/16] fix typos in torch/ (torch/_inductor/) (#156313) 2025-06-23 02:57:12 +00:00
scheduler.py Revert "[PT2][fusion] ban fusions with large accumulated reads (#157563)" 2025-07-15 16:35:55 +00:00
script.ld
select_algorithm.py multi-kernel matmuls based on varying hint sizes (#156628) 2025-07-12 15:08:21 +00:00
sizevars.py multi-kernel matmuls based on varying hint sizes (#156628) 2025-07-12 15:08:21 +00:00
standalone_compile.py [standalone_compile] Fix single Tensor outputs from split_module (#157803) 2025-07-10 12:49:03 +00:00
subgraph_lowering.py Fix XPU CI UT test_circular_dependencies (#158189) 2025-07-13 09:30:57 +00:00
template_heuristics.py multi-kernel matmuls based on varying hint sizes (#156628) 2025-07-12 15:08:21 +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 Revert "[simple_fsdp][inductor_collectives] rewrite reorder_collectives, sink_waits_iterative (#158062)" 2025-07-15 16:31:13 +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 Inductor logging + analysis of torch.profile (#149697) 2025-07-07 22:13:34 +00:00