3 commits

Author	SHA1	Message	Date
Bin Bao	787ded5103	Add lazy::Shape::numel() (#68314) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/68314 Add a convenience to lazy::Shape for counting the number of elements (by multiplying out the dimensions). This is a method on Tensor, and in switching other lazy tensor shape utils to use aten shape inference, we need numel counts. Test Plan: add unit tests Reviewed By: alanwaketan Differential Revision: D32409138 fbshipit-source-id: 3ae725300f8826d38e45412f46501d5e5f776fb2	2021-11-29 08:38:09 -08:00
Jiewen Tan	6011c35a79	[LTC] Upstream class BackendDevice (#68027) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/68027 This commit upstreams class BackendDevice to the master, which is a backend specific representation of the actual hardware, for instances, CPU, GPU, or TPU. This concept is important for backend like XLA where it needs to tell the actual hardware type from the c10::DeviceType::Lazy virtual device during both IR constructions and lowerings. Test Plan: ./build/bin/test_lazy --gtest_filter=BackendDeviceTest.* Reviewed By: wconstab Differential Revision: D32261838 Pulled By: alanwaketan fbshipit-source-id: 579c3fc5f9da7847c887a383c6047e8ecb9cc5bc	2021-11-10 07:05:43 -08:00
Jiewen Tan	8bed46ef38	[WIP][LTC] Upstream class Shape (#67672) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/67672 This commit Upstreams class Shape from lazy_tensor_staging branch. Test Plan: WIP. Reviewed By: malfet Differential Revision: D32095478 Pulled By: alanwaketan fbshipit-source-id: 61611b12fc079b195833b5b22a6cf73c0935b8b9	2021-11-04 14:12:03 -07:00