File tree Expand file tree Collapse file tree 2 files changed +9
-1
lines changed Expand file tree Collapse file tree 2 files changed +9
-1
lines changed Original file line number Diff line number Diff line change 13
13
try :
14
14
from executorch .extension .pybindings .portable_lib import ( # @manual
15
15
_load_for_executorch_from_buffer ,
16
+ Verification ,
16
17
)
17
18
except ImportError as e :
18
19
logger .warning (f"{ e = } " )
@@ -39,7 +40,9 @@ def graph_module(self) -> None:
39
40
40
41
def run_artifact (self , inputs ):
41
42
inputs_flattened , _ = tree_flatten (inputs )
42
- executorch_module = _load_for_executorch_from_buffer (self .buffer )
43
+ executorch_module = _load_for_executorch_from_buffer (
44
+ self .buffer , program_verification = Verification .Minimal
45
+ )
43
46
executorch_output = copy .deepcopy (
44
47
executorch_module .run_method ("forward" , tuple (inputs_flattened ))
45
48
)
Original file line number Diff line number Diff line change 3
3
import re
4
4
import time
5
5
import unittest
6
+ import warnings
6
7
7
8
from datetime import timedelta
8
9
from typing import Any
@@ -249,6 +250,10 @@ def build_test_filter(args: argparse.Namespace) -> TestFilter:
249
250
def runner_main ():
250
251
args = parse_args ()
251
252
253
+ # Suppress deprecation warnings for export_for_training, as it generates a
254
+ # lot of log spam. We don't really need the warning here.
255
+ warnings .simplefilter ("ignore" , category = FutureWarning )
256
+
252
257
begin_test_session (args .report )
253
258
254
259
if len (args .suite ) > 1 :
You can’t perform that action at this time.
0 commit comments