frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

remove allow-untyped-defs for torch/_C/_lazy.pyi (#143370)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/143370
Approved by: https://github.com/aorenste, https://github.com/desertfire
ghstack dependencies: #143366
This commit is contained in:
bobrenjc93
2024-12-17 07:17:52 -08:00
committed by PyTorch MergeBot
parent 487343346e
commit 313b9964ae
1 changed files with 9 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
torch/_C/_lazy.pyi
View File

@ -1,10 +1,9 @@
# mypy: allow-untyped-defs
from torch import Tensor
# 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 _mark_step(device: str, devices: list[str], wait: bool) -> None: ...
def _wait_device_ops(devices: list[str]) -> None: ...
def _reset_metrics() -> None: ...
def _counter_names() -> list[str]: ...
def _counter_value(name: str) -> int: ...
def _metrics_report() -> str: ...
@ -14,14 +13,14 @@ def _sync_multi(
devices: list[str],
wait: bool = True,
sync_ltc_data: bool = True,
): ...
) -> None: ...
def _get_tensor_id(tensor: Tensor) -> int: ...
def _get_tensors_text(tensors: list[Tensor]) -> str: ...
def _get_tensors_dot(tensors: list[Tensor]) -> str: ...
def _get_tensors_backend(tensors: list[Tensor]) -> str: ...
def _get_force_fallback() -> str: ...
def _set_force_fallback(newval: str): ...
def _clear_ir_cache(): ...
def _dump_ir_cache(filename: str): ...
def _set_reuse_ir(val: bool): ...
def _get_default_device_type(): ...
def _set_force_fallback(newval: str) -> None: ...
def _clear_ir_cache() -> None: ...
def _dump_ir_cache(filename: str) -> None: ...
def _set_reuse_ir(val: bool) -> None: ...
def _get_default_device_type() -> str: ...