pytorch/torch/_inductor/codegen
2025-02-07 21:21:21 +00:00
..
aoti_runtime
cuda [inductor] Fix test error test_force_cutlass_backend_aoti_cexpr_codegen (#146564) 2025-02-06 23:02:41 +00:00
rocm [inductor] Add typing to common.CSE (#145993) 2025-02-04 16:05:39 +00:00
xpu [inductor] Add types to DeviceOpOverrides (#145913) 2025-02-01 16:33:49 +00:00
__init__.py
aoti_hipify_utils.py
block_analysis.py
common.py Expand inductor codegen dtype asserts, fix scan (#146067) 2025-02-07 06:35:47 +00:00
cpp.py [inductor] Refactor op handlers part 1 (#146235) 2025-02-04 23:35:53 +00:00
cpp_bmm_template.py
cpp_flex_attention_template.py
cpp_gemm_template.py
cpp_grouped_gemm_template.py [inductor] Finish typing common.py (#146225) 2025-02-04 23:35:33 +00:00
cpp_micro_gemm.py [CPUInductor] Fix SVE256 detection (#146207) 2025-02-01 18:51:34 +00:00
cpp_prefix.h
cpp_template.py
cpp_template_kernel.py [inductor] Add typing to common.KernelArgs (#145916) 2025-02-04 16:05:39 +00:00
cpp_utils.py cpp_wrapper: enable all CPU repro tests (#145655) 2025-02-04 22:05:59 +00:00
cpp_wrapper_cpu.py Revert "[while_loop][inductor] support sym expression as cond_fn output (#146222)" 2025-02-07 16:19:41 +00:00
cpp_wrapper_cpu_array_ref.py
cpp_wrapper_gpu.py
cpu_device_op_overrides.py [inductor] Add types to DeviceOpOverrides (#145913) 2025-02-01 16:33:49 +00:00
cuda_combined_scheduling.py [BE] Type annotate wrapper_benchmark.py and cuda_combined_scheduling.py (#145542) 2025-01-30 03:53:52 +00:00
debug_utils.py
halide.py Revert "[inductor] Refactor op handlers part 5 (#146257)" 2025-02-05 23:20:38 +00:00
memory_planning.py
mps.py [mps] Implement support for sinc() operator (inductor and eager). (#146539) 2025-02-06 16:37:27 +00:00
mps_device_op_overrides.py [inductor] Add types to DeviceOpOverrides (#145913) 2025-02-01 16:33:49 +00:00
multi_kernel.py
simd.py [inductor] Add typing to common.CSE (#145993) 2025-02-04 16:05:39 +00:00
simd_kernel_features.py
triton.py [inductor] Use index_dtype (int32/int64 depending on size) for argmax accumulators (#146651) 2025-02-07 21:21:21 +00:00
triton_combo_kernel.py [inductor] Add typing to common.KernelArgs (#145916) 2025-02-04 16:05:39 +00:00
triton_split_scan.py [inductor] Add typing to common.CSE (#145993) 2025-02-04 16:05:39 +00:00
triton_utils.py [inductor][5/N] triton support post-#5512, fix 1 and None handling (#145515) 2025-02-01 02:11:48 +00:00
wrapper.py Revert "[while_loop][inductor] support sym expression as cond_fn output (#146222)" 2025-02-07 16:19:41 +00:00