Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[AOTI] Disable stack allocation when there is a fallback op (#122367)
Summary: Stack allocation is disabled when there is an aten fallback op, see https://github.com/pytorch/pytorch/blob/c84f81b395fff969bbd2f784efad8ab1a8aa52de/torch/_inductor/codegen/cpp_wrapper_cpu.py#L974. But we need to do the same where is a custom op fallback. Test Plan: CI Reviewed By: mikekgfb Differential Revision: D55149369 Pull Request resolved: #122367 Approved by: https://github.com/mikekgfb
- Loading branch information