feat: Add _mm_cvtps_epi32 and _mm_cvtss_f32

pattonkan · Sep 5, 2024 · 1487321 · 1487321
1 parent c90c592
commit 1487321
Show file tree

Hide file tree

Showing 2 changed files with 52 additions and 48 deletions.
diff --git a/sse2rvv.h b/sse2rvv.h
@@ -1485,7 +1485,10 @@ FORCE_INLINE __m128i _mm_cvtepu8_epi64(__m128i a) {
 
 // FORCE_INLINE __m128 _mm_cvtpi8_ps (__m64 a) {}
 
-// FORCE_INLINE __m128i _mm_cvtps_epi32 (__m128 a) {}
+FORCE_INLINE __m128i _mm_cvtps_epi32(__m128 a) {
+  vfloat32m1_t _a = vreinterpretq_m128_f32(a);
+  return vreinterpretq_i32_m128i(__riscv_vfcvt_x_f_v_i32m1(_a, 4));
+}
 
 // FORCE_INLINE __m128d _mm_cvtps_pd (__m128 a) {}
 
@@ -1531,7 +1534,10 @@ FORCE_INLINE __m128i _mm_cvtepu8_epi64(__m128i a) {
 
 // FORCE_INLINE __m128i _mm_cvtsi64x_si128 (__int64 a) {}
 
-// FORCE_INLINE float _mm_cvtss_f32 (__m128 a) {}
+FORCE_INLINE float _mm_cvtss_f32(__m128 a) {
+  vfloat32m1_t _a = vreinterpretq_m128_f32(a);
+  return (float)__riscv_vfmv_f_s_f32m1_f32(_a);
+}
 
 // FORCE_INLINE __m128d _mm_cvtss_sd (__m128d a, __m128 b) {}
 

diff --git a/tests/impl.cpp b/tests/impl.cpp
@@ -1793,18 +1793,16 @@ result_t test_mm_cvtsi64_ss(const SSE2RVV_TEST_IMPL &impl, uint32_t iter) {
 }
 
 result_t test_mm_cvtss_f32(const SSE2RVV_TEST_IMPL &impl, uint32_t iter) {
-  // #ifdef ENABLE_TEST_ALL
-  //   const float *_a = impl.test_cases_float_pointer1;
-  //
-  //   float f = _a[0];
-  //
-  //   __m128 a = load_m128(_a);
-  //   float c = _mm_cvtss_f32(a);
-  //
-  //   return f == c ? TEST_SUCCESS : TEST_FAIL;
-  // #else
+#ifdef ENABLE_TEST_ALL
+  const float *_a = impl.test_cases_float_pointer1;
+  float f = _a[0];
+  __m128 a = load_m128(_a);
+  float c = _mm_cvtss_f32(a);
+
+  return f == c ? TEST_SUCCESS : TEST_FAIL;
+#else
   return TEST_UNIMPL;
-  // #endif  // ENABLE_TEST_ALL
+#endif // ENABLE_TEST_ALL
 }
 
 result_t test_mm_cvtss_si32(const SSE2RVV_TEST_IMPL &impl, uint32_t iter) {
@@ -5094,42 +5092,42 @@ result_t test_mm_cvtpi32_pd(const SSE2RVV_TEST_IMPL &impl, uint32_t iter) {
 }
 
 result_t test_mm_cvtps_epi32(const SSE2RVV_TEST_IMPL &impl, uint32_t iter) {
-  // #ifdef ENABLE_TEST_ALL
-  //   const float *_a = impl.test_cases_float_pointer1;
-  //   __m128 a = load_m128(_a);
-  //   int32_t d[4];
-  //   switch (iter & 0x3) {
-  //   case 0:
-  //     _MM_SET_ROUNDING_MODE(_MM_ROUND_NEAREST);
-  //     for (uint32_t i = 0; i < 4; i++) {
-  //       d[i] = (int32_t)(bankers_rounding(_a[i]));
-  //     }
-  //     break;
-  //   case 1:
-  //     _MM_SET_ROUNDING_MODE(_MM_ROUND_DOWN);
-  //     for (uint32_t i = 0; i < 4; i++) {
-  //       d[i] = (int32_t)(floorf(_a[i]));
-  //     }
-  //     break;
-  //   case 2:
-  //     _MM_SET_ROUNDING_MODE(_MM_ROUND_UP);
-  //     for (uint32_t i = 0; i < 4; i++) {
-  //       d[i] = (int32_t)(ceilf(_a[i]));
-  //     }
-  //     break;
-  //   case 3:
-  //     _MM_SET_ROUNDING_MODE(_MM_ROUND_TOWARD_ZERO);
-  //     for (uint32_t i = 0; i < 4; i++) {
-  //       d[i] = (int32_t)(_a[i]);
-  //     }
-  //     break;
-  //   }
-  //
-  //   __m128i ret = _mm_cvtps_epi32(a);
-  //   return VALIDATE_INT32_M128(ret, d);
-  // #else
+#ifdef ENABLE_TEST_ALL
+  const float *_a = impl.test_cases_float_pointer1;
+  __m128 a = load_m128(_a);
+  int32_t d[4];
+  switch (iter & 0x3) {
+  case 0:
+    _MM_SET_ROUNDING_MODE(_MM_ROUND_NEAREST);
+    for (uint32_t i = 0; i < 4; i++) {
+      d[i] = (int32_t)(bankers_rounding(_a[i]));
+    }
+    break;
+  case 1:
+    _MM_SET_ROUNDING_MODE(_MM_ROUND_DOWN);
+    for (uint32_t i = 0; i < 4; i++) {
+      d[i] = (int32_t)(floorf(_a[i]));
+    }
+    break;
+  case 2:
+    _MM_SET_ROUNDING_MODE(_MM_ROUND_UP);
+    for (uint32_t i = 0; i < 4; i++) {
+      d[i] = (int32_t)(ceilf(_a[i]));
+    }
+    break;
+  case 3:
+    _MM_SET_ROUNDING_MODE(_MM_ROUND_TOWARD_ZERO);
+    for (uint32_t i = 0; i < 4; i++) {
+      d[i] = (int32_t)(_a[i]);
+    }
+    break;
+  }
+
+  __m128i ret = _mm_cvtps_epi32(a);
+  return VALIDATE_INT32_M128(ret, d);
+#else
   return TEST_UNIMPL;
-  // #endif  // ENABLE_TEST_ALL
+#endif // ENABLE_TEST_ALL
 }
 
 result_t test_mm_cvtps_pd(const SSE2RVV_TEST_IMPL &impl, uint32_t iter) {