pytorch/torch/_inductor/codegen
Bin Bao a597a00c87 [AOTI][refactor][3/n] Declare python_kernel_name and cpp_kernel_name in ExternKernel (#115972)
Summary: Both ExternKernelAlloc and ExternKernelOut need the two fields, so declaring them in the base class. Also add cpp codegen for IndexPutFallback and InplaceBernoulliFallback in this PR.

This is a reland of https://github.com/pytorch/pytorch/pull/115831

Differential Revision: [D52290900](https://our.internmc.facebook.com/intern/diff/D52290900)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/115972
Approved by: https://github.com/chenyang78
2023-12-20 03:22:03 +00:00
..
aoti_runtime [AOTInductor] Add updaing constant buffer to active buffer. (#116001) 2023-12-18 11:49:03 +00:00
cuda [inductor] Remove more type: ignore comments (#114162) 2023-11-28 06:45:55 +00:00
__init__.py
common.py add support for dynamic shapes in round (#115259) 2023-12-19 15:45:50 +00:00
cpp_prefix.h [inductor cpp] vectorize embedding lookup (#114062) 2023-11-22 11:19:42 +00:00
cpp.py add support for dynamic shapes in round (#115259) 2023-12-19 15:45:50 +00:00
cuda_combined_scheduling.py [Inductor CUTLASS backend] Epilogue fusion codegen (Step 1) (#110890) 2023-11-06 19:42:10 +00:00
memory_planning.py Properly type CachedFunction & rename to CachedMethod (#114161) 2023-11-25 01:30:23 +00:00
triton_foreach.py [Reland2] [inductor][BE] split triton_meta and inductor_meta (#112351) 2023-11-02 00:40:12 +00:00
triton_utils.py [AOTI] Support ReinterpretView in abi mode (#114169) 2023-11-21 17:08:00 +00:00
triton.py add support for dynamic shapes in round (#115259) 2023-12-19 15:45:50 +00:00
wrapper.py [AOTI][refactor][3/n] Declare python_kernel_name and cpp_kernel_name in ExternKernel (#115972) 2023-12-20 03:22:03 +00:00