Resolve Operator fallback to method allocator if temp alloc fails (#20981)#20981
Resolve Operator fallback to method allocator if temp alloc fails (#20981)#20981rstehle wants to merge 2 commits into
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/20981
Note: Links to docs will display an error until the docs builds have been completed.
|
|
@rstehle has exported this pull request. If you are a Meta employee, you can view the originating Diff in D109855815. |
This PR needs a
|
…torch#20981) Summary: Currently, the resolve_operator step during method_init will fall back to method allocator if temp allocator is zero or not present. This change adds logic to fallback to method allocator if temp allocator is present, but insufficient in size for the tensor meta. Reviewed By: Andrew-github-user, rascani Differential Revision: D109855815
2da43a9 to
186e3df
Compare
…torch#20981) Summary: Currently, the resolve_operator step during method_init will fall back to method allocator if temp allocator is zero or not present. This change adds logic to fallback to method allocator if temp allocator is present, but insufficient in size for the tensor meta. Reviewed By: Andrew-github-user, rascani Differential Revision: D109855815
186e3df to
67aa6c0
Compare
|
/easycla |
|
/easycla |
Summary:
Currently, the resolve_operator step during method_init will fall back to method allocator if temp allocator is zero or not present.
This change adds logic to fallback to method allocator if temp allocator is present, but insufficient in size for the tensor meta.
Reviewed By: Andrew-github-user, rascani
Differential Revision: D109855815