Skip to content

[Backend Tester] Reduce log verbosity / spam #13312

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 6 commits into
base: gh/GregoryComer/122/head
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 4 additions & 1 deletion backends/test/harness/stages/serialize.py
Original file line number Diff line number Diff line change
Expand Up @@ -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=}")
Expand All @@ -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))
)
Expand Down
5 changes: 5 additions & 0 deletions backends/test/suite/runner.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
import re
import time
import unittest
import warnings

from datetime import timedelta
from typing import Any
Expand Down Expand Up @@ -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:
Expand Down
Loading