pytorch/torch/_inductor
2024-05-13 17:36:52 +00:00
..
codegen [inductor] Make load_mask() codegen determinstic (#126017) 2024-05-13 17:36:52 +00:00
fx_passes [inductor] refactor: device dispatch inside do_bench (#125736) 2024-05-09 23:50:02 +00:00
kernel [inductor][cpp] GEMM template (infra and fp32) (#124021) 2024-05-12 07:46:44 +00:00
runtime Add script for removing Inductor dependencies from Inductor generated code (#125811) 2024-05-10 00:00:25 +00:00
__init__.py [aoti] Change aot_compile callsites (#122225) 2024-03-29 21:34:20 +00:00
autotune_process.py [inductor] refactor: device dispatch inside do_bench (#125736) 2024-05-09 23:50:02 +00:00
bounds.py Some minor type stub improvements (#118529) 2024-02-04 00:19:00 +00:00
codecache.py [inductor] autotune benchmark support for cpu (#125159) 2024-05-09 00:28:49 +00:00
comm_analysis.py Remove Inductor IRs for legacy functional collectives (#124992) 2024-05-05 19:49:58 +00:00
comms.py [BE]: Improve exception typing. Remove NOQAs (#125535) 2024-05-08 14:07:13 +00:00
compile_fx.py Disable fb_memcache for MTIA (#125658) 2024-05-07 07:00:26 +00:00
config.py [inductor][cpp] GEMM template (infra and fp32) (#124021) 2024-05-12 07:46:44 +00:00
constant_folding.py [inductor] Handle meta tensor ops in graph (#123786) 2024-04-12 19:03:13 +00:00
cudagraph_trees.py [BE]: Try TCH autofixes on torch/ (#125536) 2024-05-05 23:13:59 +00:00
cudagraph_utils.py [cudagraphs] add cudagraph_skips counter (#124804) 2024-04-26 03:22:29 +00:00
debug.py [inductor] Add FileLock around V.debug.copy (#122665) 2024-03-28 03:17:33 +00:00
decomposition.py [Inductor] [Quant] Enable lowering of quant per tensor and refactor quant pattern (#124041) 2024-05-09 08:40:44 +00:00
dependencies.py [Inductor] log fusion failure due to index mismatch (#124986) 2024-05-07 02:29:00 +00:00
exc.py [HOP][inductor] Add higher order associative scan operator (#119430) 2024-04-23 14:40:13 +00:00
freezing.py Use graph.find_nodes in inductor (#122256) 2024-04-07 18:51:14 +00:00
fx_utils.py Rebind and refresh unbacked bindings in FakeTensorUpdater (#124314) 2024-04-25 02:08:55 +00:00
graph.py [AOTI] Use random inputs to autotune the backward pass (#125291) 2024-05-10 20:13:34 +00:00
hooks.py Enable local_partial_types (#118467) 2024-01-28 13:38:22 +00:00
index_propagation.py [inductor] Use python constants in IndexPropagation (#122031) 2024-03-21 21:53:22 +00:00
inductor_prims.py Retry: Low mem max_pool2d_with_indices (#122832) 2024-05-08 19:37:08 +00:00
ir.py [inductor][cpp] GEMM template (infra and fp32) (#124021) 2024-05-12 07:46:44 +00:00
lowering.py Revert "update pointwise cat heuristics (#125772)" 2024-05-11 15:27:44 +00:00
metrics.py [Inductor] log fusion failure due to index mismatch (#124986) 2024-05-07 02:29:00 +00:00
mkldnn_lowerings.py [inductor][cpp] GEMM template (infra and fp32) (#124021) 2024-05-12 07:46:44 +00:00
ops_handler.py [HOP][inductor] Add higher order associative scan operator (#119430) 2024-04-23 14:40:13 +00:00
optimize_indexing.py [BE]: Improve exception typing. Remove NOQAs (#125535) 2024-05-08 14:07:13 +00:00
pattern_matcher.py [BE]: Improve exception typing. Remove NOQAs (#125535) 2024-05-08 14:07:13 +00:00
quantized_lowerings.py [AOTInductor] Support quantized linear on CPU with fbgemm (#123069) 2024-04-01 09:15:05 +00:00
scheduler.py Fix multi template debug trace (#125703) 2024-05-08 16:31:18 +00:00
script.ld Place .lrodata later in the binary (#117575) 2024-01-18 17:58:18 +00:00
select_algorithm.py [inductor][cpp] GEMM template (infra and fp32) (#124021) 2024-05-12 07:46:44 +00:00
sizevars.py [Inductor] log fusion failure due to index mismatch (#124986) 2024-05-07 02:29:00 +00:00
subgraph_lowering.py [HOP][inductor] Support pytrees as associative_scan input (#122137) 2024-05-06 11:29:28 +00:00
test_case.py [inductor] Keep inductor cache for tests when TORCH_COMPILE_DEBUG is specified (#124755) 2024-04-23 20:22:55 +00:00
test_operators.py
utils.py [inductor][cpp] GEMM template (infra and fp32) (#124021) 2024-05-12 07:46:44 +00:00
virtualized.py [EASY] Move OpsHandler and MockHandler to their own file (#119851) 2024-02-15 18:54:41 +00:00
wrapper_benchmark.py [inductor] refactor: device dispatch inside do_bench (#125736) 2024-05-09 23:50:02 +00:00