pytorch
36180c13 - [static runtime] aten::to copy out variant (#52343)

Commit
3 years ago
[static runtime] aten::to copy out variant (#52343) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/52343 aten::to returns self when the TensorOptions match and copy is set to false. For static runtime, always copy. There isn't a separate op for aten::to copy, but instead the same function with different arguments. Test Plan: On AdFinder local_ro: Before: 0.896742 0.00824827 ms. 0.92773%. aten::to (5 nodes) After: 0.88233 0.0056607 ms. 0.644675%. aten::to (5 nodes) buck test mode/opt caffe2/benchmarks/static_runtime:static_runtime_cpptest Reviewed By: hlu1 Differential Revision: D26477980 fbshipit-source-id: 8e8448092adff38c141af1ce27a10acd39c07dd1
Author
Parents
Loading