File size: 448 Bytes
9c6594c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
import torch._C._lazy
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: str) -> None:
"""Set the config used to force LTC fallback"""
torch._C._lazy._set_force_fallback(configval)
def set_reuse_ir(val: bool) -> None:
"""Set the config to reuse IR nodes for faster tracing"""
torch._C._lazy._set_reuse_ir(val)
|