pytorch/torch/_inductor/codegen
2024-07-31 16:49:21 +00:00
..
aoti_runtime [Inductor] Enable ABI-compatible mode for cpp-wrapper JIT (#121309) 2024-03-07 14:22:06 +00:00
cuda [BE][Easy][16/19] enforce style for empty lines in import segments in torch/_i*/ (#129768) 2024-07-20 16:20:58 +00:00
rocm [BE][Easy][16/19] enforce style for empty lines in import segments in torch/_i*/ (#129768) 2024-07-20 16:20:58 +00:00
xpu Flip default value for mypy disallow_untyped_defs [2/11] (#127839) 2024-06-08 18:23:08 +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
codegen_device_driver.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 Revert "Migrate Inductor scheduler, dependencies, ir, and codegen/common to use OrderedSet (#130004)" 2024-07-31 16:49:21 +00:00
cpp_gemm_template.py [BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765) 2024-07-31 10:42:50 +00:00
cpp_micro_gemm.py [inductor][cpp][gemm] optimize arbitrary N in packed gemm template (#130690) 2024-07-20 06:30:15 +00:00
cpp_prefix.h [inductor][cpp][gemm] improve thread blocking heuristics (#131024) 2024-07-24 00:36:29 +00:00
cpp_template_kernel.py Cpp wrapper: set args to CppWrapperKernelArgs in cpp template kernel (#129557) 2024-07-29 04:01:17 +00:00
cpp_template.py [inductor] turn on enable_kernel_profile on Windows. (#132025) 2024-07-30 03:02:09 +00:00
cpp_utils.py [BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765) 2024-07-31 10:42:50 +00:00
cpp_wrapper_cpu.py Back out "[1/2] PT2 Inductor ComboKernels - Foreach cases (#124969)" (#132065) 2024-07-29 22:48:29 +00:00
cpp_wrapper_cuda.py Back out "[1/2] PT2 Inductor ComboKernels - Foreach cases (#124969)" (#132065) 2024-07-29 22:48:29 +00:00
cpp.py [inductor] turn on enable_kernel_profile on Windows. (#132025) 2024-07-30 03:02:09 +00:00
cuda_combined_scheduling.py Back out "[1/2] PT2 Inductor ComboKernels - Foreach cases (#124969)" (#132065) 2024-07-29 22:48:29 +00:00
halide.py Revert "Migrate Inductor scheduler, dependencies, ir, and codegen/common to use OrderedSet (#130004)" 2024-07-31 16:49:21 +00:00
memory_planning.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
multi_kernel.py Revert "Migrate Inductor scheduler, dependencies, ir, and codegen/common to use OrderedSet (#130004)" 2024-07-31 16:49:21 +00:00
simd.py Revert "Migrate Inductor scheduler, dependencies, ir, and codegen/common to use OrderedSet (#130004)" 2024-07-31 16:49:21 +00:00
triton_foreach.py Back out "[1/2] PT2 Inductor ComboKernels - Foreach cases (#124969)" (#132065) 2024-07-29 22:48:29 +00:00
triton_split_scan.py Revert "Migrate Inductor scheduler, dependencies, ir, and codegen/common to use OrderedSet (#130004)" 2024-07-31 16:49:21 +00:00
triton_utils.py Annotate graph.py (#131400) 2024-07-23 07:04:12 +00:00
triton.py Revert "Migrate Inductor scheduler, dependencies, ir, and codegen/common to use OrderedSet (#130004)" 2024-07-31 16:49:21 +00:00
wrapper.py Back out "[1/2] PT2 Inductor ComboKernels - Foreach cases (#124969)" (#132065) 2024-07-29 22:48:29 +00:00