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

remove allow-untyped-defs from torch/_lazy/config.py (#143603)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/143603
Approved by: https://github.com/aorenste
This commit is contained in:
bobrenjc93
2024-12-19 16:15:31 -08:00
committed by PyTorch MergeBot
parent cb4e9888df
commit 78d294379a
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
torch/_lazy/config.py
View File

@ -1,17 +1,16 @@
# mypy: allow-untyped-defs
import torch._C._lazy
def get_force_fallback():
def get_force_fallback() -> str:
"""Get the config used to force LTC fallback"""
return torch._C._lazy._get_force_fallback()
def set_force_fallback(configval):
def set_force_fallback(configval: str) -> None:
"""Set the config used to force LTC fallback"""
torch._C._lazy._set_force_fallback(configval)
def set_reuse_ir(val: bool):
def set_reuse_ir(val: bool) -> None:
"""Set the config to reuse IR nodes for faster tracing"""
torch._C._lazy._set_reuse_ir(val)