arm neon sm3: check constant range

simd-everywhere · Mar 20, 2024 · 3d34fcd · 3d34fcd
1 parent 339ffe4
commit 3d34fcd
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/simde/arm/neon/sm3.h b/simde/arm/neon/sm3.h
@@ -63,7 +63,9 @@ simde_vsm3ss1q_u32(simde_uint32x4_t n, simde_uint32x4_t m, simde_uint32x4_t a) {
 
 SIMDE_FUNCTION_ATTRIBUTES
 simde_uint32x4_t
-simde_vsm3tt1aq_u32(simde_uint32x4_t a, simde_uint32x4_t b, simde_uint32x4_t c, const int imm2) {
+simde_vsm3tt1aq_u32(simde_uint32x4_t a, simde_uint32x4_t b, simde_uint32x4_t c, const int imm2)
+  SIMDE_REQUIRE_CONSTANT_RANGE(imm2, 0, 3)
+{
     simde_uint32x4_private
       r_,
       a_ = simde_uint32x4_to_private(a),
@@ -91,7 +93,9 @@ simde_vsm3tt1aq_u32(simde_uint32x4_t a, simde_uint32x4_t b, simde_uint32x4_t c,
 
 SIMDE_FUNCTION_ATTRIBUTES
 simde_uint32x4_t
-simde_vsm3tt1bq_u32(simde_uint32x4_t a, simde_uint32x4_t b, simde_uint32x4_t c, const int imm2) {
+simde_vsm3tt1bq_u32(simde_uint32x4_t a, simde_uint32x4_t b, simde_uint32x4_t c, const int imm2)
+  SIMDE_REQUIRE_CONSTANT_RANGE(imm2, 0, 3)
+{
     simde_uint32x4_private
       r_,
       a_ = simde_uint32x4_to_private(a),
@@ -119,7 +123,9 @@ simde_vsm3tt1bq_u32(simde_uint32x4_t a, simde_uint32x4_t b, simde_uint32x4_t c,
 
 SIMDE_FUNCTION_ATTRIBUTES
 simde_uint32x4_t
-simde_vsm3tt2aq_u32(simde_uint32x4_t a, simde_uint32x4_t b, simde_uint32x4_t c, const int imm2) {
+simde_vsm3tt2aq_u32(simde_uint32x4_t a, simde_uint32x4_t b, simde_uint32x4_t c, const int imm2)
+  SIMDE_REQUIRE_CONSTANT_RANGE(imm2, 0, 3)
+{
     simde_uint32x4_private
       r_,
       a_ = simde_uint32x4_to_private(a),
@@ -146,7 +152,9 @@ simde_vsm3tt2aq_u32(simde_uint32x4_t a, simde_uint32x4_t b, simde_uint32x4_t c,
 
 SIMDE_FUNCTION_ATTRIBUTES
 simde_uint32x4_t
-simde_vsm3tt2bq_u32(simde_uint32x4_t a, simde_uint32x4_t b, simde_uint32x4_t c, const int imm2) {
+simde_vsm3tt2bq_u32(simde_uint32x4_t a, simde_uint32x4_t b, simde_uint32x4_t c, const int imm2)
+  SIMDE_REQUIRE_CONSTANT_RANGE(imm2, 0, 3)
+{
     simde_uint32x4_private
       r_,
       a_ = simde_uint32x4_to_private(a),