mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
65b9778d30
Pull Request resolved: https://github.com/pytorch/pytorch/pull/76488 Approved by: https://github.com/wconstab, https://github.com/JackCaoG
14 lines
417 B
Python
14 lines
417 B
Python
import torch._C._lazy
|
|
|
|
def get_force_fallback():
|
|
"""Get the config used to force LTC fallback"""
|
|
return torch._C._lazy._get_force_fallback()
|
|
|
|
def set_force_fallback(configval):
|
|
"""Set the config used to force LTC fallback"""
|
|
torch._C._lazy._set_force_fallback(configval)
|
|
|
|
def set_reuse_ir(val: bool):
|
|
"""Set the config to reuse IR nodes for faster tracing"""
|
|
torch._C._lazy._set_reuse_ir(val)
|