pytorch
c3147504 - [JIT] enable profiling optional tensors (#70532)

Commit
2 years ago
[JIT] enable profiling optional tensors (#70532) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/70532 This adds profiling to Optional[Tensor] types First, in profiling_record.cpp, profiling nodes are added to Optional[Tensor] inputs. The nodes record (a) whether or not any `None` types are encountered, and (b) of the Tensor types, what's the most specific type matching all of non-null tensors that were encoutered (shape, dtype, etc.) In tensorexpr_fuser, when specializing types based on the profiled information, an Optional[Tensor] type will always be Optional[], but the Tensor type contained in the optional type can be specialized (e.g. `Optional[Float(2x2x2, cpu, etc)]`) Test Plan: Imported from OSS Reviewed By: albanD Differential Revision: D33714748 Pulled By: davidberard98 fbshipit-source-id: 93c819054450de7ac84b112de1012c0c12e34120 (cherry picked from commit 21cfd8012396cf4cc8437cb1dbf6b3846720d1b9)
Author
Committer
Parents
Loading