pytorch/torch/_inductor/codegen
2024-10-18 23:05:54 +00:00
..
aoti_runtime [aoti][inplace] Support skipping model buffers (#136770) 2024-09-30 18:28:42 +00:00
cuda [inductor] Generalize WorkspaceArg for graph-level semaphores (#138170) 2024-10-18 23:05:54 +00:00
rocm [inductor] Generalize WorkspaceArg for graph-level semaphores (#138170) 2024-10-18 23:05:54 +00:00
xpu [Inductor] Generalize device guard codegen for cpp_wrapper mode. (#134761) 2024-09-10 10:11:52 +00:00
__init__.py
aoti_hipify_utils.py [BE][Easy][16/19] enforce style for empty lines in import segments in torch/_i*/ (#129768) 2024-07-20 16:20:58 +00:00
common.py [inductor] Generalize WorkspaceArg for graph-level semaphores (#138170) 2024-10-18 23:05:54 +00:00
cpp_gemm_template.py Port Inductor dataclasses to be kw_only (#137768) 2024-10-14 10:33:43 +00:00
cpp_micro_gemm.py [Inductor][CPP] Optimize WOQ INT8 wgt dequant in AMX GEMM template (#136630) 2024-09-26 08:41:58 +00:00
cpp_prefix.h Extend vectorization with SVE(ARM) with Torch Compile (Inductor) (#134672) 2024-10-10 13:20:40 +00:00
cpp_template_kernel.py Refactored debug_extra to be on ChoiceCaller (and called description) (#137857) 2024-10-15 00:48:14 +00:00
cpp_template.py [AOTI] Remove the non-ABI-compatible mode (part 1) (#138009) 2024-10-17 02:48:26 +00:00
cpp_utils.py Moved some of Inductor IR nodes to be frozen (#137859) 2024-10-17 18:04:45 +00:00
cpp_wrapper_cpu_array_ref.py [inductor] Generalize WorkspaceArg for graph-level semaphores (#138170) 2024-10-18 23:05:54 +00:00
cpp_wrapper_cpu.py [AOTI] Remove the non-ABI-compatible mode (part 1) (#138009) 2024-10-17 02:48:26 +00:00
cpp_wrapper_gpu.py [inductor] Generalize WorkspaceArg for graph-level semaphores (#138170) 2024-10-18 23:05:54 +00:00
cpp.py Port Inductor dataclasses to be kw_only (#137768) 2024-10-14 10:33:43 +00:00
cpu_device_op_overrides.py Add Triton CPU as an Inductor backend (#133408) 2024-09-30 20:24:52 +00:00
cuda_combined_scheduling.py [BE]: Update mypy to 1.11.2 (#133816) 2024-09-16 19:44:11 +00:00
debug_utils.py [AOTI] Remove the non-ABI-compatible mode (part 1) (#138009) 2024-10-17 02:48:26 +00:00
halide.py Add Triton CPU as an Inductor backend (#133408) 2024-09-30 20:24:52 +00:00
memory_planning.py [inductor] Generalize WorkspaceArg for graph-level semaphores (#138170) 2024-10-18 23:05:54 +00:00
multi_kernel.py remove fast_flush arguments (#135387) 2024-09-13 08:13:46 +00:00
simd.py Remove an unused variable in _inductor/codegen/simd.py (#138000) 2024-10-16 13:54:21 +00:00
triton_combo_kernel.py [inductor] Generalize WorkspaceArg for graph-level semaphores (#138170) 2024-10-18 23:05:54 +00:00
triton_split_scan.py [inductor] Delete dead code and lints (#137753) 2024-10-11 18:55:08 +00:00
triton_utils.py [inductor] Generalize WorkspaceArg for graph-level semaphores (#138170) 2024-10-18 23:05:54 +00:00
triton.py [inductor] Generalize WorkspaceArg for graph-level semaphores (#138170) 2024-10-18 23:05:54 +00:00
wrapper.py [inductor] Generalize WorkspaceArg for graph-level semaphores (#138170) 2024-10-18 23:05:54 +00:00