You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
17 lines
448 B
17 lines
448 B
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)
|