mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
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 |
||
|---|---|---|
| .. | ||
| aoti_runtime | ||
| cuda | ||
| __init__.py | ||
| common.py | ||
| cpp_prefix.h | ||
| cpp.py | ||
| cuda_combined_scheduling.py | ||
| memory_planning.py | ||
| triton_foreach.py | ||
| triton_utils.py | ||
| triton.py | ||
| wrapper.py | ||