pytorch/torch/_inductor/codegen
2024-08-01 15:26:45 +00:00
..
aoti_runtime Revert "[AOTI] Fix bfloat16 in CPU (#132150)" 2024-08-01 03:35:39 +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 Add None return type to init (#132335) 2024-08-01 15:26:45 +00:00
cpp_gemm_template.py typing ir.py - part 1 (#131845) 2024-07-31 17:37:14 +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 Revert "[AOTI] Fix bfloat16 in CPU (#132150)" 2024-08-01 03:35:39 +00:00
cpp_wrapper_cuda.py Add None return type to init (#132335) 2024-08-01 15:26:45 +00:00
cpp.py [inductor] cpp codegen alignas for all OSs. (#132387) 2024-08-01 14:30: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 Migrate Inductor scheduler, dependencies, ir, and codegen/common to use OrderedSet (#130004) 2024-08-01 04:37:15 +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 Migrate Inductor scheduler, dependencies, ir, and codegen/common to use OrderedSet (#130004) 2024-08-01 04:37:15 +00:00
simd.py Migrate Inductor scheduler, dependencies, ir, and codegen/common to use OrderedSet (#130004) 2024-08-01 04:37:15 +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 Migrate Inductor scheduler, dependencies, ir, and codegen/common to use OrderedSet (#130004) 2024-08-01 04:37:15 +00:00
triton_utils.py Annotate graph.py (#131400) 2024-07-23 07:04:12 +00:00
triton.py Add None return type to init (#132335) 2024-08-01 15:26:45 +00:00
wrapper.py Back out "[1/2] PT2 Inductor ComboKernels - Foreach cases (#124969)" (#132065) 2024-07-29 22:48:29 +00:00