Add neon-i8x32 and neon-16x16 to ISPCTarget

aneshlya · aneshlya · commit a87045969dbc · 2025-01-24T17:46:15.000Z
diff --git a/src/target_enums.cpp b/src/target_enums.cpp
@@ -183,12 +183,18 @@ ISPCTarget operator++(ISPCTarget &target, int dummy) {
     static_assert(static_cast<underlying>(ISPCTarget::neon_i8x16) ==
                       static_cast<underlying>(ISPCTarget::avx512spr_x64) + 1,
                   "Enum ISPCTarget is not sequential");
-    static_assert(static_cast<underlying>(ISPCTarget::neon_i16x8) ==
+    static_assert(static_cast<underlying>(ISPCTarget::neon_i8x32) ==
                       static_cast<underlying>(ISPCTarget::neon_i8x16) + 1,
                   "Enum ISPCTarget is not sequential");
-    static_assert(static_cast<underlying>(ISPCTarget::neon_i32x4) ==
+    static_assert(static_cast<underlying>(ISPCTarget::neon_i16x8) ==
+                      static_cast<underlying>(ISPCTarget::neon_i8x32) + 1,
+                  "Enum ISPCTarget is not sequential");
+    static_assert(static_cast<underlying>(ISPCTarget::neon_i16x16) ==
                       static_cast<underlying>(ISPCTarget::neon_i16x8) + 1,
                   "Enum ISPCTarget is not sequential");
+    static_assert(static_cast<underlying>(ISPCTarget::neon_i32x4) ==
+                      static_cast<underlying>(ISPCTarget::neon_i16x16) + 1,
+                  "Enum ISPCTarget is not sequential");
     static_assert(static_cast<underlying>(ISPCTarget::neon_i32x8) ==
                       static_cast<underlying>(ISPCTarget::neon_i32x4) + 1,
                   "Enum ISPCTarget is not sequential");
@@ -402,8 +408,12 @@ ISPCTarget ParseISPCTarget(std::string target) {
         return ISPCTarget::avx512spr_x64;
     } else if (target == "neon-i8x16") {
         return ISPCTarget::neon_i8x16;
+    } else if (target == "neon-i8x32") {
+        return ISPCTarget::neon_i8x32;
     } else if (target == "neon-i16x8") {
         return ISPCTarget::neon_i16x8;
+    } else if (target == "neon-i16x16") {
+        return ISPCTarget::neon_i16x16;
     } else if (target == "neon-i32x4" || target == "neon") {
         return ISPCTarget::neon_i32x4;
     } else if (target == "neon-i32x8") {
@@ -581,8 +591,12 @@ std::string ISPCTargetToString(ISPCTarget target) {
         return "avx512spr-x64";
     case ISPCTarget::neon_i8x16:
         return "neon-i8x16";
+    case ISPCTarget::neon_i8x32:
+        return "neon-i8x32";
     case ISPCTarget::neon_i16x8:
         return "neon-i16x8";
+    case ISPCTarget::neon_i16x16:
+        return "neon-i16x16";
     case ISPCTarget::neon_i32x4:
         return "neon-i32x4";
     case ISPCTarget::neon_i32x8:
diff --git a/src/target_enums.h b/src/target_enums.h
@@ -74,7 +74,9 @@ enum class ISPCTarget {
     avx512spr_x32,
     avx512spr_x64,
     neon_i8x16,
+    neon_i8x32,
     neon_i16x8,
+    neon_i16x16,
     neon_i32x4,
     neon_i32x8,
     wasm_i32x4,
