pytorch/test/inductor
Sam Larsen fc1105b282 [inductor] Implement Fx graph caching to improve warm compilation time. (#103453)
Summary: Implement an on-disk cache to save and reuse compiled FX Graphs. This implementation does not handle tensors with symbolic shapes. This needs to be done in a follow-up PR.

Test Plan:
* New unit tests exercising saving and load from the cache.
* New unit tests to exercise the cache key calculations.
* Ran several benchmarks to see cache hit and resulting compilation times.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/103453
Approved by: https://github.com/eellison, https://github.com/Chillee
2023-10-11 14:39:14 +00:00
..
cpp
extension_backends Extend Inductor to support the third-party backend (#106874) 2023-08-16 04:11:36 +00:00
__init__.py
indirect_assert_helper.py
minifier_smoke.py
opinfo_harness.py
test_aot_inductor.py [inductor] Add AOTI ABI shim function for torch.nonzero (#110766) 2023-10-07 08:32:27 +00:00
test_binary_folding.py [Inductor][FX]Support efficient conv bn eval (#108757) 2023-09-20 08:10:02 +00:00
test_codecache.py [inductor] Implement Fx graph caching to improve warm compilation time. (#103453) 2023-10-11 14:39:14 +00:00
test_codegen_triton.py [inductor] triton_utils.config_of: check for divisibility by 16, even when expr is not an Integer (#105743) 2023-07-24 22:41:50 +00:00
test_compiled_autograd.py retain_graph=True in compiled_autograd (#110367) 2023-10-06 08:22:10 +00:00
test_compiled_optimizers.py Add compiled optimizer test for nadam (#109548) 2023-09-19 22:54:36 +00:00
test_config.py [dynamo] Add BACKEND_MATCH guard to detect and recompile when backend changes (#107337) 2023-09-14 15:49:30 +00:00
test_coordinate_descent_tuner.py [inductur] coordesc tuner bug fix with no_x_dim kernel (#104692) 2023-07-06 17:47:02 +00:00
test_cpp_wrapper.py [aotinductor] Fix a missing schema issue for repeat_interleave (#110105) 2023-09-29 23:01:37 +00:00
test_cpu_repro.py Inductor cpp wrapper: support MkldnnRnnLayer (#107858) 2023-09-29 00:22:42 +00:00
test_cuda_repro.py Change flash attention outputs to be SymInt instead of int (#110533) 2023-10-05 01:00:07 +00:00
test_cudacodecache.py Disable cutlass tests in fbcode (#109241) 2023-09-14 01:41:10 +00:00
test_cudagraph_trees.py Extend storage access error api for untyped_storage() (#109750) 2023-09-25 17:51:27 +00:00
test_custom_lowering.py [inductor] Optimize away zero-element loads (#107074) 2023-08-12 07:58:14 +00:00
test_custom_post_grad_passes.py inductor: add custom pass hooks in post_grad_passes (#108615) 2023-09-11 04:13:32 +00:00
test_dependencies.py [inductor] Implement bucketize() for dependencies.py (#105102) 2023-07-17 19:15:00 +00:00
test_efficient_conv_bn_eval.py [Inductor][FX]support nn.Linear nn.ConvTransposeNd for efficient_conv_bn_eval (#109722) 2023-09-23 01:12:34 +00:00
test_extension_backend.py Fix vscode test discovery (#107404) 2023-08-18 03:50:46 +00:00
test_foreach.py fix(inductor): Increase coverage of Inductor ATen lowering (#110473) 2023-10-04 23:40:46 +00:00
test_fp8.py Basic fp8 support in Inductor (#109168) 2023-09-23 04:41:41 +00:00
test_fused_attention.py fix sfdp patern 13 accuracy issue (#110001) 2023-09-26 15:23:45 +00:00
test_fx_fusion.py Enable additional inductor test suites on ROCm (#102270) 2023-06-22 00:36:35 +00:00
test_group_batch_fusion.py [inductor][fx passes] batch tanh in pre grad (#107881) 2023-08-25 03:02:30 +00:00
test_indexing.py [inductor] Add min/max to index propagation pass (#105020) 2023-07-12 19:03:01 +00:00
test_inductor_freezing.py Fix spelling / capitalization in freezing.py error message (#109347) 2023-09-18 18:12:20 +00:00
test_inductor_utils.py [Inductor CUTLASS backend] Step 3: autotune_process, and CUDABenchmarkRequest (#107901) 2023-09-12 17:44:36 +00:00
test_inplacing_pass.py Add reinplacing pass for scatters + incremental fake tensor updating (#106192) 2023-08-30 20:41:37 +00:00
test_kernel_benchmark.py [BE]: Update Ruff to 0.0.280 (#105724) 2023-07-22 23:03:34 +00:00
test_layout_optim.py [ROCm] enable additional inductor/dynamo UTs (#104624) 2023-07-11 20:44:02 +00:00
test_max_autotune.py Revert "[inductor] fix a max-autotune rng state related bug (#109828)" 2023-09-23 22:35:37 +00:00
test_minifier_isolate.py Move has_triton to top level triton utils so that dynamo can also access (#109832) 2023-09-22 19:33:41 +00:00
test_minifier.py Move has_triton to top level triton utils so that dynamo can also access (#109832) 2023-09-22 19:33:41 +00:00
test_mkldnn_pattern_matcher.py [Quant][Inductor] Enable quantization dynamic batch size support (#108550) 2023-09-19 08:30:16 +00:00
test_mmdecomp.py Move has_triton to top level triton utils so that dynamo can also access (#109832) 2023-09-22 19:33:41 +00:00
test_pattern_matcher.py [inductor] Make sure unfuse_addmm and addmm patterns don't overlap (#110235) 2023-09-29 19:35:29 +00:00
test_perf.py [Inductor] Allow matmul to have flexiable layout when we are not autotuning (#110726) 2023-10-07 04:08:37 +00:00
test_profiler.py Move has_triton to top level triton utils so that dynamo can also access (#109832) 2023-09-22 19:33:41 +00:00
test_select_algorithm.py [inductor] Parallelize Max Autotune step 2: Use multiple GPUs (#109127) 2023-09-14 00:37:39 +00:00
test_smoke.py [ROCm] enable cudagraph inductor UTs on ROCm (#105662) 2023-08-01 20:55:27 +00:00
test_snode_runtime.py Add reinplacing pass for scatters + incremental fake tensor updating (#106192) 2023-08-30 20:41:37 +00:00
test_split_cat_fx_passes.py Back out "[Inductor][FX passes] Remove config.split_cat_fx_passes & A… (#104370) 2023-07-01 00:44:46 +00:00
test_standalone_compile.py [BE] Enable ruff's UP rules and autoformat inductor/ (#105431) 2023-07-19 13:45:00 +00:00
test_torchinductor_codegen_dynamic_shapes.py [Reland] [Inductor] Break the loop fusion when node2 depends on node1 mutations (#110677) 2023-10-11 00:26:45 +00:00
test_torchinductor_dynamic_shapes.py [dynamo] Trace through builtin abs (#110398) 2023-10-03 19:25:37 +00:00
test_torchinductor_opinfo.py [Reland] [Inductor] Break the loop fusion when node2 depends on node1 mutations (#110677) 2023-10-11 00:26:45 +00:00
test_torchinductor.py Move export.constrain_as_* to torch._constrain_as_* (#110757) 2023-10-11 02:37:55 +00:00
test_triton_heuristics.py [reland][inductor] make thread order consistent with loop order (#107902) 2023-08-26 02:56:20 +00:00
test_triton_wrapper.py [BE]: Update Ruff to 0.0.280 (#105724) 2023-07-22 23:03:34 +00:00