OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
5ad5f40283
pytorch/torch/_inductor/codegen
History
Bin Bao 5ad5f40283 [AOTI][reland] Create another wrapper class to handle ArrayRef (#136461) 
Summary: Create another wrapper codegen class to handle ArrayRef for CPU. The goal is to simplify the regular cpp wrapper codegen logic and the generated cpp code.

Test Plan: CI

Differential Revision: [D63300361](https://our.internmc.facebook.com/intern/diff/D63300361)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/136461
Approved by: https://github.com/angelayi, https://github.com/chenyang78
ghstack dependencies: #136062
2024-09-25 14:00:09 +00:00
..
aoti_runtime [AOTI] Fix bfloat16 in CPU (#132150) 2024-08-01 22:26:30 +00:00
cuda [Inductor] Rename WrapperCodeGen to PythonWrapperCodegen (#136062) 2024-09-24 21:02:51 +00:00
rocm [Inductor] Rename WrapperCodeGen to PythonWrapperCodegen (#136062) 2024-09-24 21:02:51 +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 [AOTI][reland] Create another wrapper class to handle ArrayRef (#136461) 2024-09-25 14:00:09 +00:00
cpp_gemm_template.py [inductor] [cpp] fix max-autotune for single-thread dynamic shapes (#136418) 2024-09-25 09:24:05 +00:00
cpp_micro_gemm.py [AOTI][reland] Fix assert_function call in cpu autotune template (#135920) 2024-09-13 12:21:57 +00:00
cpp_prefix.h [AOTI][reland] Fix assert_function call in cpu autotune template (#135920) 2024-09-13 12:21:57 +00:00
cpp_template_kernel.py Revert "Add Triton CPU as an Inductor backend (#133408)" 2024-09-16 18:33:33 +00:00
cpp_template.py [AOTI][reland] Fix assert_function call in cpu autotune template (#135920) 2024-09-13 12:21:57 +00:00
cpp_utils.py [inductor] [cpp] add index check when fusing epilogue with GEMM template (#135661) 2024-09-24 05:25:28 +00:00
cpp_wrapper_cpu_array_ref.py [AOTI][reland] Create another wrapper class to handle ArrayRef (#136461) 2024-09-25 14:00:09 +00:00
cpp_wrapper_cpu.py [Inductor] Rename WrapperCodeGen to PythonWrapperCodegen (#136062) 2024-09-24 21:02:51 +00:00
cpp_wrapper_gpu.py [aoti] Fix workspace generation for triton (#135552) 2024-09-22 04:51:37 +00:00
cpp.py [inductor] [cpp] fix the check of template_buffer_has_other_users if no epilogue_nodes (#136518) 2024-09-25 10:25:07 +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][Tooling] Filter out kernels based off lowercase names (#135395) 2024-09-20 21:56:08 +00:00
halide.py Revert "Add Triton CPU as an Inductor backend (#133408)" 2024-09-16 18:33:33 +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 remove fast_flush arguments (#135387) 2024-09-13 08:13:46 +00:00
simd.py [inductor] Split reduction loops when there is no shared reads (#134307) 2024-09-12 09:45:08 +00:00
triton_combo_kernel.py [pytorch][PR] [inductor] More fixes on the keys of constants and signature dictionaries (#136514) 2024-09-25 09:34:14 +00:00
triton_split_scan.py [aoti] Fix workspace generation for triton (#135552) 2024-09-22 04:51:37 +00:00
triton_utils.py [pytorch][PR] [inductor] More fixes on the keys of constants and signature dictionaries (#136514) 2024-09-25 09:34:14 +00:00
triton.py [pytorch][PR] [inductor] More fixes on the keys of constants and signature dictionaries (#136514) 2024-09-25 09:34:14 +00:00
wrapper.py [pytorch][PR] [inductor] More fixes on the keys of constants and signature dictionaries (#136514) 2024-09-25 09:34:14 +00:00