pytorch/torch/_inductor/codegen
2024-05-13 17:36:52 +00:00
..
aoti_runtime [Inductor] Enable ABI-compatible mode for cpp-wrapper JIT (#121309) 2024-03-07 14:22:06 +00:00
cuda Get PT2 Cutlass backend working under fbcode [take 2] (#125688) 2024-05-08 16:44:49 +00:00
xpu [Inductor Intel GPU backend Upstream] Add Inductor Intel GPU backend. (#121895) 2024-04-05 09:05:11 +00:00
__init__.py
aoti_hipify_utils.py [5/x][AMD][Lowering Enablement] Hipifying aoti code_wrapper (#124241) 2024-04-19 18:57:38 +00:00
codegen_device_driver.py [5/x][AMD][Lowering Enablement] Hipifying aoti code_wrapper (#124241) 2024-04-19 18:57:38 +00:00
common.py Compute bounds for the variables created during codegen (#123100) 2024-05-08 08:14:06 +00:00
cpp_gemm_template.py [inductor][cpp] GEMM template (infra and fp32) (#124021) 2024-05-12 07:46:44 +00:00
cpp_micro_gemm.py [inductor][cpp] GEMM template (infra and fp32) (#124021) 2024-05-12 07:46:44 +00:00
cpp_prefix.h [inductor][cpp] GEMM template (infra and fp32) (#124021) 2024-05-12 07:46:44 +00:00
cpp_template_kernel.py [inductor][cpp] GEMM template (infra and fp32) (#124021) 2024-05-12 07:46:44 +00:00
cpp_template.py [inductor][cpp] GEMM template (infra and fp32) (#124021) 2024-05-12 07:46:44 +00:00
cpp_utils.py [inductor][cpp] GEMM template (infra and fp32) (#124021) 2024-05-12 07:46:44 +00:00
cpp_wrapper_cpu.py [AOTI] Fix convolution_backward (#125730) 2024-05-10 20:13:34 +00:00
cpp_wrapper_cuda.py [AOTI] Fix 'int' object is not subscriptable (#125731) 2024-05-11 18:12:46 +00:00
cpp.py [inductor][cpp] GEMM template (infra and fp32) (#124021) 2024-05-12 07:46:44 +00:00
cuda_combined_scheduling.py [Inductor cutlass backend] Remove epilogue nodes from Kernel call (#124929) 2024-05-02 13:02:31 +00:00
memory_planning.py Fix global flake8 issues (#124771) 2024-04-26 15:35:53 +00:00
multi_kernel.py [inductor] refactor: device dispatch inside do_bench (#125736) 2024-05-09 23:50:02 +00:00
triton_foreach.py [inductor] Remove usage of device_interface from _inductor.runtime (#124592) 2024-04-30 16:54:16 +00:00
triton_split_scan.py [inductor] Remove config check for 3D tiling (#124569) 2024-04-22 18:46:40 +00:00
triton_utils.py [inductor] Specialize on unguarded alignment of example inputs (#123319) 2024-04-25 22:28:15 +00:00
triton.py [inductor] Make load_mask() codegen determinstic (#126017) 2024-05-13 17:36:52 +00:00
wrapper.py [Inductor cutlass backend] Enabled nonzero workspace and Cutlass StreamK (#125406) 2024-05-05 15:28:45 +00:00