Skip to content

Commit b991dcb

Browse files
[ExecuTorch] Arm Ethos Bento Kernel
Pull Request resolved: #8405 Add a bento-kernel with basic functionality to lower PyTorch programs to ET PTE with Ethos cmd stream in it. * Known issues * We haven't yet buckified `tosa_reference_model` - those tests will fail * No e2e tests yet, need to wire-up FVP with this kernel ghstack-source-id: 265991547 @exported-using-ghexport Differential Revision: [D69496532](https://our.internmc.facebook.com/intern/diff/D69496532/) Co-authored-by: Digant Desai <[email protected]>
1 parent c8311e6 commit b991dcb

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)