xref: /aosp_15_r20/external/pytorch/torch/_lazy/config.py (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1# mypy: allow-untyped-defs
2import torch._C._lazy
3
4
5def get_force_fallback():
6    """Get the config used to force LTC fallback"""
7    return torch._C._lazy._get_force_fallback()
8
9
10def set_force_fallback(configval):
11    """Set the config used to force LTC fallback"""
12    torch._C._lazy._set_force_fallback(configval)
13
14
15def set_reuse_ir(val: bool):
16    """Set the config to reuse IR nodes for faster tracing"""
17    torch._C._lazy._set_reuse_ir(val)
18