config.py 448 B

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