Skip to content

Commit b02cf9d

Browse files
committed
Update default model build spec
1 parent 7e6138e commit b02cf9d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

model/aot_model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,8 +25,8 @@ def forward(self, x: torch.Tensor, y: torch.Tensor) -> torch.Tensor:
2525
# explanation of its flags: https://gitlab.arm.com/artificial-intelligence/ethos-u/ethos-u-vela/-/blob/main/OPTIONS.md
2626
compile_spec = EthosUCompileSpec(
2727
target="ethos-u85-256",
28-
system_config="Ethos_U55_High_End_Embedded",
29-
memory_mode="Shared_Sram",
28+
system_config="Ethos_U85_SYS_DRAM_Mid",
29+
memory_mode="Sram_Only",
3030
extra_flags=["--output-format=raw", "--debug-force-regor"]
3131
)
3232

0 commit comments

Comments
 (0)