Skip to content

Commit bb91715

Browse files
committed
[ExecuTorch] Arm Ethos Bento Kernel
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 Differential Revision: [D69496532](https://our.internmc.facebook.com/intern/diff/D69496532/) ghstack-source-id: 265972752 Pull Request resolved: #8405
1 parent eacbeb7 commit bb91715

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

backends/arm/test/runner_utils.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,13 +10,19 @@
1010
import shutil
1111
import subprocess
1212
import tempfile
13+
from logging import logger
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+
try:
22+
import tosa_reference_model
23+
except ImportError:
24+
logger.warning("tosa_reference_model not found, can't run reference model tests")
25+
tosa_reference_model = None
2026
from executorch.backends.arm.arm_backend import get_tosa_version, is_tosa
2127

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

0 commit comments

Comments
 (0)