Adding all project files
This commit is contained in:
parent
6c9e127bdc
commit
cd4316ad0f
42289 changed files with 8009643 additions and 0 deletions
16
venv/Lib/site-packages/torch/_lazy/config.py
Normal file
16
venv/Lib/site-packages/torch/_lazy/config.py
Normal file
|
@ -0,0 +1,16 @@
|
|||
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)
|
Loading…
Add table
Add a link
Reference in a new issue