Skip to content

Commit 4cba9df

Browse files
Enable training IR
1 parent 97a4600 commit 4cba9df

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

exir/__init__.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,8 @@
2929
from executorch.exir.tracer import ExirDynamoConfig
3030
from torch.export import ExportedProgram, ExportGraphSignature
3131

32+
import torch
33+
3234
Value = Any
3335

3436
__all__ = [
@@ -55,3 +57,8 @@
5557
"load",
5658
"save",
5759
]
60+
61+
def func(*args, **kwargs):
62+
return True
63+
64+
torch._utils_internal.export_training_ir_rollout_check = func

0 commit comments

Comments
 (0)