File tree Expand file tree Collapse file tree 2 files changed +10
-2
lines changed Expand file tree Collapse file tree 2 files changed +10
-2
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
@@ -250,7 +251,11 @@ def build_test_filter(args: argparse.Namespace) -> TestFilter:
250
251
def runner_main ():
251
252
args = parse_args ()
252
253
253
- begin_test_session ()
254
+ # Suppress deprecation warnings for export_for_training, as it generates a
255
+ # lot of log spam. We don't really need the warning here.
256
+ warnings .simplefilter ("ignore" , category = FutureWarning )
257
+
258
+ begin_test_session (args .report )
254
259
255
260
if len (args .suite ) > 1 :
256
261
raise NotImplementedError ("TODO Support multiple suites." )
You can’t perform that action at this time.
0 commit comments