pytorch
902b4dba - Change capture_scalar_outputs to use SymInt/SymFloat rather than Tensor to model scalars (#93150)

Commit
1 year ago
Change capture_scalar_outputs to use SymInt/SymFloat rather than Tensor to model scalars (#93150) Previously, Dynamo faked support for item() when `capture_scalar_outputs` was True by representing it internally as a Tensor. With dynamic shapes, this is no longer necessary; we can represent it directly as a SymInt/SymFloat. Do so. Doing this requires you to use dynamic shapes; in principle we could support scalar outputs WITHOUT dynamic shapes but I won't do this unless someone hollers for it. Signed-off-by: Edward Z. Yang <ezyang@meta.com> Differential Revision: [D42885775](https://our.internmc.facebook.com/intern/diff/D42885775) Pull Request resolved: https://github.com/pytorch/pytorch/pull/93150 Approved by: https://github.com/voznesenskym
Author
Committer
Parents
Loading