mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
ff206ed09e
Summary: This adds a minimal set of python bindings for lazy tensor and the torchscript backend. It targets the APIs that are used by the `test_ts_opinfo.py` test (which it also lands, from lazy_tensor_staging, where it is [lazy_tensor_core/test/test_lazy.py](https://github.com/pytorch/pytorch/blob/lazy_tensor_staging/lazy_tensor_core/test/test_lazy.py)). We should land more python bindings obviously. I just wanted to focus on a minimal set that can also be tested, and use it to agree on how we organize the bindings, then others could easily contribute bindings on top of this infrastructure. cc JackCaoG Pull Request resolved: https://github.com/pytorch/pytorch/pull/74508 Reviewed By: pbelevich Differential Revision: D35032152 Pulled By: wconstab fbshipit-source-id: 526505ab355b7ad27037ece0ff814b2a4b69f1e2 (cherry picked from commit b4f73dd147472cb38003204aff228087c0230fda)
8 lines
245 B
Python
8 lines
245 B
Python
from typing import List
|
|
|
|
#defined in torch/csrc/lazy/python/init.cpp
|
|
def _mark_step(device: str, devices: List[str], wait: bool): ...
|
|
def _wait_device_ops(devices: List[str]): ...
|
|
def _reset_metrics(): ...
|
|
def _counter_names() -> List[str]: ...
|