diff --git a/backends/test/harness/stages/serialize.py b/backends/test/harness/stages/serialize.py index 9d0bded0483..a5be1631d98 100644 --- a/backends/test/harness/stages/serialize.py +++ b/backends/test/harness/stages/serialize.py @@ -13,6 +13,7 @@ try: from executorch.extension.pybindings.portable_lib import ( # @manual _load_for_executorch_from_buffer, + Verification, ) except ImportError as e: logger.warning(f"{e=}") @@ -39,7 +40,9 @@ def graph_module(self) -> None: def run_artifact(self, inputs): inputs_flattened, _ = tree_flatten(inputs) - executorch_module = _load_for_executorch_from_buffer(self.buffer) + executorch_module = _load_for_executorch_from_buffer( + self.buffer, program_verification=Verification.Minimal + ) executorch_output = copy.deepcopy( executorch_module.run_method("forward", tuple(inputs_flattened)) ) diff --git a/backends/test/suite/runner.py b/backends/test/suite/runner.py index 4999779b3c9..eea1ce6b404 100644 --- a/backends/test/suite/runner.py +++ b/backends/test/suite/runner.py @@ -3,6 +3,7 @@ import re import time import unittest +import warnings from datetime import timedelta from typing import Any @@ -249,6 +250,10 @@ def build_test_filter(args: argparse.Namespace) -> TestFilter: def runner_main(): args = parse_args() + # Suppress deprecation warnings for export_for_training, as it generates a + # lot of log spam. We don't really need the warning here. + warnings.simplefilter("ignore", category=FutureWarning) + begin_test_session(args.report) if len(args.suite) > 1: