pytorch
f11f0e4a - [inductor] Handle nested tuple/list output in fallback kernel (#88495)

Commit
2 years ago
[inductor] Handle nested tuple/list output in fallback kernel (#88495) Summary: Currently fallback kernel in inductor assumes its output is either a tensor or a tuple/list of tensors. This PR makes it handle more generic output data structure. Pull Request resolved: https://github.com/pytorch/pytorch/pull/88495 Approved by: https://github.com/jansel
Author
Committer
Parents
Loading