Skip to content

Commit 428f561

Browse files
authored
[ExecuTorch] Arm Ethos Bento Kernel
Differential Revision: D69496532 Pull Request resolved: #8405
1 parent eacbeb7 commit 428f561

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

backends/arm/test/runner_utils.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,13 +10,20 @@
1010
import shutil
1111
import subprocess
1212
import tempfile
13+
1314
from pathlib import Path
1415

1516
from typing import cast, Dict, List, Literal, Optional, Tuple
1617

1718
import numpy as np
1819
import torch
19-
import tosa_reference_model
20+
21+
logger = logging.getLogger(__name__)
22+
try:
23+
import tosa_reference_model
24+
except ImportError:
25+
logger.warning("tosa_reference_model not found, can't run reference model tests")
26+
tosa_reference_model = None
2027
from executorch.backends.arm.arm_backend import get_tosa_version, is_tosa
2128

2229
from executorch.backends.arm.test.conftest import is_option_enabled

0 commit comments

Comments
 (0)