[CUDA, test-suite] More test cases for rint() and nearint()

[tra]

Summary

Provides coverage for the bug fixed in D85236

External/CUDA/math_h.cu

__device__ void test_nearbyint()
     static_assert((std::is_same<decltype(nearbyint((long long)0)), double>::value), "");
     static_assert((std::is_same<decltype(nearbyint((unsigned long long)0)), double>::value), "");
     static_assert((std::is_same<decltype(nearbyint((double)0)), double>::value), "");
     static_assert((std::is_same<decltype(nearbyintf(0)), float>::value), "");
     static_assert((std::is_same<decltype(nearbyint(Ambiguous())), Ambiguous>::value), "");
     assert(nearbyint(V(1)) == 1);
     assert(nearbyint(V(1.)) == 1);
     assert(nearbyint(V(1.f)) == 1);
+    // There are more checks in test_rint(). rint and nearbyint behave the same
+    // way on the GPU, so we only test them in one place.
 }
 
 __device__ void test_nextafter()
 {
     static_assert((std::is_same<decltype(nextafter((float)0, (float)0)), float>::value), "");
     static_assert((std::is_same<decltype(nextafter((bool)0, (float)0)), double>::value), "");
     static_assert((std::is_same<decltype(nextafter((unsigned short)0, (double)0)), double>::value), "");
     static_assert((std::is_same<decltype(nextafter((float)0, (unsigned int)0)), double>::value), "");
__device__ void test_remquo()
     assert(std::remquo(V(0.5), 1., &ip) == 0.5);
     assert(std::remquo(V(0.5), 1.f, &ip) == 0.5);
 
     assert(std::remquo(V(0.5f), 1, &ip) == 0.5);
     assert(std::remquo(V(0.5f), 1., &ip) == 0.5);
     assert(std::remquo(V(0.5f), 1.f, &ip) == 0.5);
 }
 
-__device__ void test_rint()
+__device__ void test_rint_nearbyint()
 {
     static_assert((std::is_same<decltype(rint((float)0)), float>::value), "");
     static_assert((std::is_same<decltype(rint((bool)0)), double>::value), "");
     static_assert((std::is_same<decltype(rint((unsigned short)0)), double>::value), "");
     static_assert((std::is_same<decltype(rint((int)0)), double>::value), "");
     static_assert((std::is_same<decltype(rint((unsigned int)0)), double>::value), "");
     static_assert((std::is_same<decltype(rint((long)0)), double>::value), "");
     static_assert((std::is_same<decltype(rint((unsigned long)0)), double>::value), "");
     static_assert((std::is_same<decltype(rint((long long)0)), double>::value), "");
     static_assert((std::is_same<decltype(rint((unsigned long long)0)), double>::value), "");
     static_assert((std::is_same<decltype(rint((double)0)), double>::value), "");
     static_assert((std::is_same<decltype(rintf(0)), float>::value), "");
     static_assert((std::is_same<decltype(rint(Ambiguous())), Ambiguous>::value), "");
-    assert(rint(V(1)) == 1);
-    assert(rint(V(1.)) == 1);
-    assert(rint(V(1.f)) == 1);
+    // Verify that rint/nearbyint produce identical correct results
+    auto check = [](double input, double fpresult) {
+      // FP rint()/nearbyint must match the expected result.
+      assert(rint(V(float(input))) == float(fpresult));
+      assert(nearbyint(V(float(input))) == float(fpresult));
+      assert(rint(V(input)) == fpresult);
+      assert(nearbyint(V(input)) == fpresult);
+      // for integral types, std::rint(input) == std::rint(double(input))
+      int iinput = input;
+      assert(std::rint(V(iinput)) == std::rint(double(V(iinput))));
+      assert(std::nearbyint(V(iinput)) == std::nearbyint(double(V(iinput))));
+    };
+    // Whole values round to themselves and do not change sign.
+    check(0.0, 0.0);
+    check(-0.0, -0.0);
+    check(1.0, 1.0);
+    check(-1.0, -1.0);
+    // Half-way values round towards nearest even number.
+    check(2.5, 2.0);
+    check(-2.5, -2.0);
+    check(3.5, 4.0);
+    check(-3.5, -4.0);
+    // Everything else is rounded towards nearest integer.
+    check(2.1, 2.0);
+    check(-2.1, -2.0);
+    check(2.7, 3.0);
+    check(-2.7, -3.0);
+    check(3.9, 4.0);
+    check(-3.9, -4.0);
 }
 
 __device__ void test_round()
 {
     static_assert((std::is_same<decltype(round((float)0)), float>::value), "");
     static_assert((std::is_same<decltype(round((bool)0)), double>::value), "");
     static_assert((std::is_same<decltype(round((unsigned short)0)), double>::value), "");
     static_assert((std::is_same<decltype(round((int)0)), double>::value), "");
__global__ void tests()
     test_logb();
     test_lrint();
     test_lround();
     test_nan();
     test_nearbyint();
     test_nextafter();
     test_remainder();
     test_remquo();
-    test_rint();
+    test_rint_nearbyint();
     test_round();
     test_scalbln();
     test_scalbn();
     test_tgamma();
     test_trunc();
 }
 
 int main() {