Skip to content

Commit 02297f9

Browse files
TEMP way to run export_for_training
ghstack-source-id: 47cb401 Pull Request resolved: #6136
1 parent 69766fb commit 02297f9

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

exir/__init__.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,9 @@
2828
from executorch.exir.serde.serialize import load, save
2929
from executorch.exir.tracer import ExirDynamoConfig
3030
from torch.export import ExportedProgram, ExportGraphSignature
31+
import torch
32+
33+
3134

3235
Value = Any
3336

@@ -55,3 +58,9 @@
5558
"load",
5659
"save",
5760
]
61+
62+
63+
def func(*args, **kwargs):
64+
return True
65+
66+
torch._utils_internal.export_training_ir_rollout_check = func

0 commit comments

Comments
 (0)