#137 Add Raspberry Pi 5 & AVH Ethos-U Series device (#145)

Nota-NetsPresso · Feb 2, 2024 · e8dc257 · e8dc257
1 parent ceb1013
commit e8dc257
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/netspresso/enums/device.py b/netspresso/enums/device.py
@@ -3,6 +3,7 @@
 
 
 class DeviceName(str, Enum):
+    RASPBERRY_PI_5 = "RaspberryPi5"
     RASPBERRY_PI_4B = "RaspberryPi4B"
     RASPBERRY_PI_3B_PLUS = "RaspberryPi3BPlus"
     RASPBERRY_PI_ZERO_W = "RaspberryPi-ZeroW"
@@ -18,10 +19,12 @@ class DeviceName(str, Enum):
     INTEL_XEON_W_2233 = "Intel-Xeon"
     ALIF_ENSEMBLE_E7_DEVKIT_GEN2 = "Ensemble-E7-DevKit-Gen2"
     RENESAS_RA8D1 = "Renesas-RA8D1"
+    ARM_ETHOS_U_SERIES = "Arm Virtual Hardware Ethos-U Series"
 
     @classmethod
     def create_literal(cls):
         return Literal[
+            "RaspberryPi5",
             "RaspberryPi4B",
             "RaspberryPi3BPlus",
             "RaspberryPi-ZeroW",
@@ -37,6 +40,7 @@ def create_literal(cls):
             "Intel-Xeon",
             "Ensemble-E7-DevKit-Gen2",
             "Renesas-RA8D1",
+            "Arm Virtual Hardware Ethos-U Series",
         ]
 
     JETSON_DEVICES = [
@@ -47,6 +51,7 @@ def create_literal(cls):
         JETSON_AGX_ORIN,
     ]
     RASPBERRY_PI_DEVICES = [
+        RASPBERRY_PI_5,
         RASPBERRY_PI_4B,
         RASPBERRY_PI_3B_PLUS,
         RASPBERRY_PI_ZERO_W,
@@ -58,12 +63,14 @@ def create_literal(cls):
     AVAILABLE_INT8_DEVICES = [
         ALIF_ENSEMBLE_E7_DEVKIT_GEN2,
         RENESAS_RA8D1,
+        RASPBERRY_PI_5,
         RASPBERRY_PI_4B,
         RASPBERRY_PI_3B_PLUS,
         RASPBERRY_PI_ZERO_W,
         RASPBERRY_PI_ZERO_2W,
+        ARM_ETHOS_U_SERIES,
     ]
-    ONLY_INT8_DEVICES = [ALIF_ENSEMBLE_E7_DEVKIT_GEN2, RENESAS_RA8D1]
+    ONLY_INT8_DEVICES = [ALIF_ENSEMBLE_E7_DEVKIT_GEN2, RENESAS_RA8D1, ARM_ETHOS_U_SERIES]
 
 
 class SoftwareVersion(str, Enum):