pytorch/torch/_inductor/codegen
2024-05-23 08:50:18 +00:00
..
aoti_runtime [Inductor] Enable ABI-compatible mode for cpp-wrapper JIT (#121309) 2024-03-07 14:22:06 +00:00
cuda typing scheduler.py [2/2]: Apply types (#126656) 2024-05-22 20:33:31 +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 Revert "[inductor][cpp] epilogue support for gemm template (#126019)" 2024-05-23 08:50:18 +00:00
cpp_gemm_template.py Revert "[inductor][cpp] epilogue support for gemm template (#126019)" 2024-05-23 08:50:18 +00:00
cpp_micro_gemm.py Revert "[inductor][cpp] bf16/fp16 gemm template computed with fp32 w/o epilogue fusion (#126068)" 2024-05-23 08:50:18 +00:00
cpp_prefix.h [inductor][cpp] GEMM template (infra and fp32) (#124021) 2024-05-23 07:39:29 +00:00
cpp_template_kernel.py Revert "[inductor][cpp] epilogue support for gemm template (#126019)" 2024-05-23 08:50:18 +00:00
cpp_template.py Revert "[inductor][cpp] epilogue support for gemm template (#126019)" 2024-05-23 08:50:18 +00:00
cpp_utils.py Revert "[inductor][cpp] bf16/fp16 gemm template computed with fp32 w/o epilogue fusion (#126068)" 2024-05-23 08:50:18 +00:00
cpp_wrapper_cpu.py typing scheduler.py [2/2]: Apply types (#126656) 2024-05-22 20:33:31 +00:00
cpp_wrapper_cuda.py [AOTI] Fix a codegen issue when .item() is used for kernel arg (#126575) 2024-05-21 18:20:20 +00:00
cpp.py Revert "[inductor][cpp] epilogue support for gemm template (#126019)" 2024-05-23 08:50:18 +00:00
cuda_combined_scheduling.py typing scheduler.py [2/2]: Apply types (#126656) 2024-05-22 20:33:31 +00:00
memory_planning.py Fix global flake8 issues (#124771) 2024-04-26 15:35:53 +00:00
multi_kernel.py typing scheduler.py [2/2]: Apply types (#126656) 2024-05-22 20:33:31 +00:00
simd.py typing scheduler.py [2/2]: Apply types (#126656) 2024-05-22 20:33:31 +00:00
triton_foreach.py typing scheduler.py [2/2]: Apply types (#126656) 2024-05-22 20:33:31 +00:00
triton_split_scan.py [halide-backend] Refactor codegen/triton.py into codegen/simd.py (#126415) 2024-05-18 02:43:42 +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] use smaller RBLOCK for expensive reduction kernels (#126477) 2024-05-22 22:47:10 +00:00
wrapper.py typing scheduler.py [2/2]: Apply types (#126656) 2024-05-22 20:33:31 +00:00