[#57693] app: Added files for testing hifive_unmatched by Kenning

Signed-off-by: Grzegorz Latosinski <[email protected]> Signed-off-by: Maciej Torhan <[email protected]>
antmicro · Apr 8, 2024 · 1c67ab7 · 1c67ab7
1 parent 4c0ee63
commit 1c67ab7
Show file tree

Hide file tree

Showing 5 changed files with 91 additions and 1 deletion.
diff --git a/app/boards/hifive_unleashed.overlay b/app/boards/hifive_unleashed.overlay
@@ -1,5 +1,10 @@
+&uart1 {
+    status = "okay";
+    current-speed = <115200>;
+};
+
 / {
     aliases {
-        kcomms = &uart0;
+        kcomms = &uart1;
     };
 };
diff --git a/app/boards/hifive_unmatched.overlay b/app/boards/hifive_unmatched.overlay
@@ -0,0 +1,10 @@
+&uart1 {
+    status = "okay";
+    current-speed = <115200>;
+};
+
+/ {
+    aliases {
+        kcomms = &uart1;
+    };
+};
diff --git a/kenning-scenarios/renode-zephyr-tflite-magic-wand-inference-hifive-unmatched.json b/kenning-scenarios/renode-zephyr-tflite-magic-wand-inference-hifive-unmatched.json
@@ -0,0 +1,44 @@
+{
+    "dataset": {
+        "type": "kenning.datasets.magic_wand_dataset.MagicWandDataset",
+        "parameters": {
+            "dataset_root": "./build/MagicWandDataset"
+        }
+    },
+    "model_wrapper": {
+        "type": "kenning.modelwrappers.classification.tflite_magic_wand.MagicWandModelWrapper",
+        "parameters": {
+            "model_path": "kenning:///models/classification/magic_wand.h5"
+        }
+    },
+    "optimizers":
+    [
+        {
+            "type": "kenning.optimizers.tflite.TFLiteCompiler",
+            "parameters":
+            {
+                "compiled_model_path": "./build/tflite-magic-wand.tflite",
+                "inference_input_type": "float32",
+                "inference_output_type": "float32"
+            }
+	}
+    ],
+    "runtime": {
+        "type": "kenning.runtimes.renode.RenodeRuntime",
+        "parameters": {
+            "runtime_binary_path": "build/zephyr/zephyr.elf",
+            "platform_resc_path": "renode/scripts/hifive_unmatched.resc",
+            "runtime_log_uart": "/tmp/uart-log",
+            "runtime_log_init_msg": "I: Inference server started",
+            "profiler_dump_path": "build/profiler.dump"
+        }
+    },
+    "protocol": {
+        "type": "kenning.protocols.uart.UARTProtocol",
+        "parameters": {
+            "port": "/tmp/uart",
+            "baudrate": 115200,
+            "endianness": "little"
+        }
+    }
+}
diff --git a/lib/kenning_inference_lib/Kconfig b/lib/kenning_inference_lib/Kconfig
@@ -43,6 +43,7 @@ config KENNING_COMMUNICATION_PROTOCOL_NONE
 config KENNING_COMMUNICATION_PROTOCOL_UART
         bool
         prompt "UART protocol"
+        select UART_SIFIVE_PORT_1 if UART_SIFIVE
 
 endchoice
 

diff --git a/renode/scripts/hifive_unmatched.resc b/renode/scripts/hifive_unmatched.resc
@@ -0,0 +1,30 @@
+mach create "hifive_unmatched"
+
+machine LoadPlatformDescription $ORIGIN/../../build/hifive_unmatched.repl
+
+# Enable UART connection
+emulation CreateUartPtyTerminal "term-log" "/tmp/uart-log"
+connector Connect sysbus.uart0 term-log
+
+emulation CreateUartPtyTerminal "term" "/tmp/uart"
+connector Connect sysbus.uart1 term
+
+sysbus.cpu0 EnableRiscvOpcodesCounting
+sysbus.cpu0 EnableVectorOpcodesCounting
+
+sysbus.cpu1 EnableVectorOpcodesCounting
+sysbus.cpu2 EnableVectorOpcodesCounting
+sysbus.cpu3 EnableVectorOpcodesCounting
+sysbus.cpu4 EnableVectorOpcodesCounting
+
+sysbus.u74_1 EnableRiscvOpcodesCounting
+sysbus.u74_2 EnableRiscvOpcodesCounting
+sysbus.u74_3 EnableRiscvOpcodesCounting
+sysbus.u74_4 EnableRiscvOpcodesCounting
+
+sysbus.u74_1 EnableVectorOpcodesCounting
+sysbus.u74_2 EnableVectorOpcodesCounting
+sysbus.u74_3 EnableVectorOpcodesCounting
+sysbus.u74_4 EnableVectorOpcodesCounting
+
+sysbus LoadELF $bin