config.py 420 B

12345678910111213141516
  1. import torch._C._lazy
  2. def get_force_fallback():
  3. """Get the config used to force LTC fallback"""
  4. return torch._C._lazy._get_force_fallback()
  5. def set_force_fallback(configval):
  6. """Set the config used to force LTC fallback"""
  7. torch._C._lazy._set_force_fallback(configval)
  8. def set_reuse_ir(val: bool):
  9. """Set the config to reuse IR nodes for faster tracing"""
  10. torch._C._lazy._set_reuse_ir(val)