pytorch/test/cpp
Jiewen Tan ab57f6d12c [LTC] Upstream utils to extract BackendDevice from at::Tensor (#70069)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/70069

This commit upstreams utils to extract BackendDevice from at::Tensor.

Test Plan: ./build/bin/test_lazy --gtest_filter=BackendDeviceTest.GetBackendDevice*

Reviewed By: samdow

Differential Revision: D33293160

Pulled By: alanwaketan

fbshipit-source-id: 78647239f90b4d04adce84ae6022b8983ad30c09
2021-12-23 12:42:03 -08:00
..
api [C++ API] Added missing nearest-exact mode and anti-alias flag (#69318) 2021-12-22 11:10:51 -08:00
c10d Make c10d tests -Werror clean (#69703) 2021-12-09 22:10:04 -08:00
common
dist_autograd Fix distributed autograd gradients synchronization (#57792) 2021-05-09 17:32:59 -07:00
jit [jit][edge] Remove usage of shared_ptr<mobile::Code>. (#68037) 2021-12-16 13:11:46 -08:00
lazy [LTC] Upstream utils to extract BackendDevice from at::Tensor (#70069) 2021-12-23 12:42:03 -08:00
lite_interpreter_runtime Back out "Revert D30710710: [Pytorch Edge] Support profiling kineto events from external source" (#66421) 2021-10-12 10:55:29 -07:00
monitor Revert D33163841: Revert D33102715: Back out "Revert D32606547: torch/monitor: add C++ events and handlers" 2021-12-16 11:12:18 -08:00
rpc Remove ProcessGroup from TensorPipeAgent initialization (#68128) 2021-11-11 12:28:55 -08:00
tensorexpr [nnc] Add a run method to TensorExprKernel that takes in output tensors (#69477) 2021-12-22 00:30:15 -08:00
__init__.py