agm

Author: jtlap

test/unit/module/core/agm.cpp

@@ -15,7 +15,7 @@
 //==================================================================================================
 //== Types tests
 //==================================================================================================
-TTS_CASE_TPL("Check return types of agm", eve::test::simd::ieee_reals)
+TTS_CASE_TPL("Check return types of agm", eve::test::simd::ieee_reals_wf16)
 <typename T>(tts::type<T>)
 {
   using v_t = eve::element_type_t<T>;
@@ -28,21 +28,33 @@ TTS_CASE_TPL("Check return types of agm", eve::test::simd::ieee_reals)
 //== agm tests
 //==================================================================================================
 TTS_CASE_WITH("Check behavior of agm(wide)",
-              eve::test::simd::ieee_reals,
+              eve::test::simd::ieee_reals_wf16,
               tts::generate(tts::randoms(1, 10), tts::randoms(1, 10)))
 <typename T>(T const& a0, T const& a1)
 {
+  using v_t = eve::element_type_t<T>;
+  auto ref = [](auto e, auto f){
+    if constexpr(sizeof(v_t) == 2)
+    {
+      auto ee = eve::convert(e, eve::as<float>());
+      auto ff = eve::convert(f, eve::as<float>());
+      auto r = eve::agm(ee, ff);
+      return eve::convert(r, eve::as<eve::float16_t>());
+    }
+    else
+    {
+      return   ((e + f) / eve::ellint_1((e-f) / (e+f))) * eve::pio_4(eve::as(e)); 
+    }
+  };
   using eve::agm;
-  TTS_RELATIVE_EQUAL(agm(a0, a1),
-                     ((a0 + a1) / eve::ellint_1((a0 - a1) / (a0 + a1))) * eve::pio_4(eve::as(a0)),
-                     0.5);
+  TTS_RELATIVE_EQUAL(agm(a0, a1), ref(a0, a1), 0.5);
 };
 
 //==================================================================================================
 //==  conditional agm tests
 //==================================================================================================
 TTS_CASE_WITH("Check behavior of  agm[cond](wide)",
-              eve::test::simd::ieee_reals,
+              eve::test::simd::ieee_reals_wf16,
               tts::generate(tts::randoms(0, 10), tts::randoms(0, 10), tts::randoms(0, 10)))
 <typename T>(T const& a0, T const& a1, T const& a2)
 {
@@ -58,7 +70,7 @@ TTS_CASE_WITH("Check behavior of  agm[cond](wide)",
 // Tests for masked agm
 //==================================================================================================
 TTS_CASE_WITH("Check behavior of eve::masked(eve::agm)(eve::wide)",
-              eve::test::simd::ieee_reals,
+              eve::test::simd::ieee_reals_wf16,
               tts::generate(tts::randoms(eve::valmin, eve::valmax),
                             tts::randoms(eve::valmin, eve::valmax),
                             tts::logicals(0, 3)))