Skip to content
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
30 changes: 25 additions & 5 deletions backends/nxp/backend/neutron_target_spec.py
Original file line number Diff line number Diff line change
Expand Up @@ -122,20 +122,40 @@ def get_name(self) -> str:

# Whether the target has subsystem (Neutron-S) or not (Neutron-C).
def is_subsystem(self) -> bool:
return self.neutron_target.subsystem
return (
self.neutron_target.npu.subsystem
if hasattr(self.neutron_target, "npu")
else self.neutron_target.subsystem
)

# Number of compute units.
def get_num_units(self) -> int:
return self.neutron_target.numUnits
return (
self.neutron_target.npu.numUnits
if hasattr(self.neutron_target, "npu")
else self.neutron_target.numUnits
)

# Number of compute pipelines.
def get_num_pipes(self) -> int:
return self.neutron_target.numPipes
return (
self.neutron_target.npu.numPipes
if hasattr(self.neutron_target, "npu")
else self.neutron_target.numPipes
)

# Number of compute MACs.
def get_num_macs(self) -> int:
return self.neutron_target.numMacs
return (
self.neutron_target.npu.numMacs
if hasattr(self.neutron_target, "npu")
else self.neutron_target.numMacs
)

# Neutron compute block hardware version.
def get_hw_version(self) -> NeutronHWVersion:
return NeutronHWVersion(self.neutron_target.hwVersion)
return NeutronHWVersion(
self.neutron_target.npu.hwVersion
if hasattr(self.neutron_target, "npu")
else self.neutron_target.hwVersion
)
2 changes: 1 addition & 1 deletion backends/nxp/requirements-eiq.txt
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
--index-url https://eiq.nxp.com/repository
eiq-neutron-sdk==3.1.3
eiq-neutron-sdk==3.2.0
eiq_nsys
Original file line number Diff line number Diff line change
Expand Up @@ -757,6 +757,10 @@ def test__fwd_impl_b(self, input_shape, out_channels, use_qat, mocker, request):
s := 1,
d := 1,
id=f"bounds of kernel height: {_conv_id(ins, oc, ks=ks, s=s, d=d)}",
marks=pytest.mark.xfail(
reason="AIR-14679",
strict=True,
),
),
pytest.param(
ins := (4, 3, 3, 8500),
Expand Down
4 changes: 2 additions & 2 deletions docs/source/backends/nxp/nxp-overview.md
Original file line number Diff line number Diff line change
Expand Up @@ -24,10 +24,10 @@ Among currently supported machine learning models are:

- [MCUXpresso IDE](https://www.nxp.com/design/design-center/software/development-software/mcuxpresso-software-and-tools-/mcuxpresso-integrated-development-environment-ide:MCUXpresso-IDE) or [MCUXpresso Visual Studio Code extension](https://www.nxp.com/design/design-center/software/development-software/mcuxpresso-software-and-tools-/mcuxpresso-for-visual-studio-code:MCUXPRESSO-VSC)
- [MCUXpresso SDK 26.06](https://mcuxpresso.nxp.com/mcuxsdk/26.06.00/html/index.html)
- eIQ Neutron SDK version 3.1.3, what you can download from eIQ PyPI:
- eIQ Neutron SDK version 3.2.0, what you can download from eIQ PyPI:

```commandline
$ pip install --index-url https://eiq.nxp.com/repository eiq-neutron-sdk==3.1.3
$ pip install --index-url https://eiq.nxp.com/repository eiq-neutron-sdk==3.2.0
```

Instead of manually installing requirements, except MCUXpresso IDE and SDK, you can use the setup script:
Expand Down
2 changes: 1 addition & 1 deletion examples/nxp/setup.sh
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@ set -u
EIQ_PYPI_URL="${EIQ_PYPI_URL:-https://eiq.nxp.com/repository}"

# Install eIQ Neutron dependencies - SDK and simulator
pip install --index-url ${EIQ_PYPI_URL} eiq-neutron-sdk==3.1.3 eiq_nsys
pip install --index-url ${EIQ_PYPI_URL} eiq-neutron-sdk==3.2.0 eiq_nsys

# Get the directory of the current script
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
Expand Down
Loading