pytorch/torch/_inductor/codegen
PyTorch MergeBot e385bf8ef8 Revert "[halide-backend] Disable split reductions for Halide (#129320)"
This reverts commit a18eb651d3.

Reverted https://github.com/pytorch/pytorch/pull/129320 on behalf of https://github.com/jeanschmidt due to This PR is breaking internal builds, please check comments on it D59204360 ([comment](https://github.com/pytorch/pytorch/pull/129320#issuecomment-2200351678))
2024-07-01 14:44:35 +00:00
..
aoti_runtime [Inductor] Enable ABI-compatible mode for cpp-wrapper JIT (#121309) 2024-03-07 14:22:06 +00:00
cuda [inductor][refactor] Unify the use of generate_kernel_call (#128467) 2024-06-19 07:47:25 +00:00
rocm [Inductor][ROCm] Composable Kernel backend for Inductor (#125453) 2024-06-25 20:54:14 +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 Fix hipify regular expression for AOTI wrapper (#128912) 2024-06-21 18:00:40 +00:00
codegen_device_driver.py Add warpSize to Device properties (#128449) 2024-07-01 09:13:32 +00:00
common.py Revert "[halide-backend] Disable split reductions for Halide (#129320)" 2024-07-01 14:44:35 +00:00
cpp_gemm_template.py [Inductor][CPP] Enable Quantized Linear GEMM Template with INT8 output and Unary Post Op (#129048) 2024-06-30 09:53:55 +00:00
cpp_micro_gemm.py [Inductor][CPP] Enable Quantized Linear GEMM Template with FP32 output (#128825) 2024-06-30 09:45:43 +00:00
cpp_prefix.h [inductor][cpp] BF16 AMX micro-gemm support (#127195) 2024-06-21 07:21:47 +00:00
cpp_template_kernel.py [inductor] unificate toolchain code. (#129816) 2024-06-29 23:21:13 +00:00
cpp_template.py [inductor][cpp] refactor CppTemplateKernel to inherit CppKernel (#129101) 2024-06-22 12:50:37 +00:00
cpp_utils.py [Inductor][CPP] Add Min/Max with VecMask (#126841) 2024-06-18 00:20:32 +00:00
cpp_wrapper_cpu.py [AOTI] Fix test_dynamic_scalar_abi_compatible_cpu_with_stack_allocation (#129173) 2024-06-28 16:27:42 +00:00
cpp_wrapper_cuda.py [AOTI] Properly indent launchKernel calls in AOTInductor (#129616) 2024-06-28 19:16:18 +00:00
cpp.py Revert "[halide-backend] Disable split reductions for Halide (#129320)" 2024-07-01 14:44:35 +00:00
cuda_combined_scheduling.py [Inductor][ROCm] Composable Kernel backend for Inductor (#125453) 2024-06-25 20:54:14 +00:00
halide.py Revert "[halide-backend] Disable split reductions for Halide (#129320)" 2024-07-01 14:44:35 +00:00
memory_planning.py Flip default value for mypy disallow_untyped_defs [2/11] (#127839) 2024-06-08 18:23:08 +00:00
multi_kernel.py [halide-backend] Add GPU support (#127506) 2024-06-29 14:06:21 +00:00
simd.py [inductor] Add lowering and codegen for aten.sort (#128458) 2024-06-26 01:36:39 +00:00
triton_foreach.py [inductor][refactor] Unify the use of generate_kernel_call (#128467) 2024-06-19 07:47:25 +00:00
triton_split_scan.py Flip default value for mypy disallow_untyped_defs [2/11] (#127839) 2024-06-08 18:23:08 +00:00
triton_utils.py Flip default value for mypy disallow_untyped_defs [2/11] (#127839) 2024-06-08 18:23:08 +00:00
triton.py [halide-backend] Add GPU support (#127506) 2024-06-29 14:06:21 +00:00
wrapper.py [halide-backend] Add GPU support (#127506) 2024-06-29 14:06:21 +00:00