Skip to content

Commit 4bdc399

Browse files
committed
Add model specs
1 parent 140e98d commit 4bdc399

File tree

2 files changed

+5
-4
lines changed

2 files changed

+5
-4
lines changed

model/aot_model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,8 +24,8 @@ def forward(self, x: torch.Tensor, y: torch.Tensor) -> torch.Tensor:
2424
# Some args are used by the Arm Vela graph compiler later in the example. Refer to Arm Vela documentation for an
2525
# explanation of its flags: https://gitlab.arm.com/artificial-intelligence/ethos-u/ethos-u-vela/-/blob/main/OPTIONS.md
2626
compile_spec = EthosUCompileSpec(
27-
target="ethos-u55-128",
28-
system_config="Ethos_U55_High_End_Embedded",
27+
target="ethos-u85-256",
28+
system_config="Ethos_U85_SYS_DRAM_Low",
2929
memory_mode="Shared_Sram",
3030
extra_flags=["--output-format=raw", "--debug-force-regor"]
3131
)

model/arm-none-eabi-gcc.cmake

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
# modified to align better with cs300 platform
2121

2222
set(TARGET_CPU
23-
"cortex-m55"
23+
"cortex-m85"
2424
CACHE STRING "Target CPU"
2525
)
2626
string(TOLOWER ${TARGET_CPU} CMAKE_SYSTEM_PROCESSOR)
@@ -60,7 +60,8 @@ string(REPLACE "cortex-m85" "cortex-m55" GCC_CPU ${GCC_CPU})
6060
add_compile_options(
6161
-mcpu=${GCC_CPU} -mthumb "$<$<CONFIG:DEBUG>:-gdwarf-3>"
6262
"$<$<COMPILE_LANGUAGE:CXX>:-fno-unwind-tables;-fno-rtti;-fno-exceptions>"
63-
-fdata-sections -ffunction-sections
63+
-fdata-sections -ffunction-sections
64+
-Oz
6465
)
6566

6667
# Compile defines

0 commit comments

Comments
 (0)