add reference tests for zero NaN behavior

Author: upsj

reference/test/matrix/csr_kernels.cpp

@@ -1,4 +1,4 @@
-// SPDX-FileCopyrightText: 2017 - 2024 The Ginkgo authors
+// SPDX-FileCopyrightText: 2017 - 2025 The Ginkgo authors
 //
 // SPDX-License-Identifier: BSD-3-Clause
 
@@ -518,6 +518,22 @@ TYPED_TEST(Csr, AppliesLinearCombinationToDenseVector)
 }
 
 
+TYPED_TEST(Csr, AppliesLinearCombinationToDenseVectorWithZeroBetaNaN)
+{
+    using Vec = typename TestFixture::Vec;
+    using T = typename TestFixture::value_type;
+    auto alpha = gko::initialize<Vec>({-1.0}, this->exec);
+    auto beta = gko::initialize<Vec>({0.0}, this->exec);
+    auto x = gko::initialize<Vec>({2.0, 1.0, 4.0}, this->exec);
+    auto y = gko::initialize<Vec>({gko::nan<T>(), gko::nan<T>()}, this->exec);
+
+    this->mtx->apply(alpha, x, beta, y);
+
+    EXPECT_EQ(y->at(0), T{-13.0});
+    EXPECT_EQ(y->at(1), T{-5.0});
+}
+
+
 TYPED_TEST(Csr, MixedAppliesLinearCombinationToDenseVector1)
 {
     // Both vectors have the same value type which differs from the matrix

reference/test/matrix/dense_kernels.cpp

@@ -230,6 +230,25 @@ TYPED_TEST(Dense, AppliesLinearCombinationToDense)
 }
 
 
+TYPED_TEST(Dense, AppliesLinearCombinationToDenseWithZeroBetaNan)
+{
+    using Mtx = typename TestFixture::Mtx;
+    using T = typename TestFixture::value_type;
+    auto alpha = gko::initialize<Mtx>({-1.0}, this->exec);
+    auto beta = gko::initialize<Mtx>({0.0}, this->exec);
+    this->mtx3->fill(gko::nan<T>());
+
+    this->mtx2->apply(alpha, this->mtx1, beta, this->mtx3);
+
+    EXPECT_EQ(this->mtx3->at(0, 0), T{0.5});
+    EXPECT_EQ(this->mtx3->at(0, 1), T{0.5});
+    EXPECT_EQ(this->mtx3->at(0, 2), T{0.5});
+    EXPECT_EQ(this->mtx3->at(1, 0), T{-1.0});
+    EXPECT_EQ(this->mtx3->at(1, 1), T{-1.0});
+    EXPECT_EQ(this->mtx3->at(1, 2), T{-1.0});
+}
+
+
 TYPED_TEST(Dense, AppliesLinearCombinationToMixedDense)
 {
     using MixedMtx = typename TestFixture::MixedMtx;

reference/test/matrix/ell_kernels.cpp

@@ -1,4 +1,4 @@
-// SPDX-FileCopyrightText: 2017 - 2024 The Ginkgo authors
+// SPDX-FileCopyrightText: 2017 - 2025 The Ginkgo authors
 //
 // SPDX-License-Identifier: BSD-3-Clause
 
@@ -244,6 +244,21 @@ TYPED_TEST(Ell, AppliesLinearCombinationToDenseVector)
 }
 
 
+TYPED_TEST(Ell, AppliesLinearCombinationToDenseVectorWithZeroBetaNaN)
+{
+    using T = typename TestFixture::value_type;
+    using Vec = typename TestFixture::Vec;
+    auto alpha = gko::initialize<Vec>({-1.0}, this->exec);
+    auto beta = gko::initialize<Vec>({0.0}, this->exec);
+    auto x = gko::initialize<Vec>({2.0, 1.0, 4.0}, this->exec);
+    auto y = gko::initialize<Vec>({gko::nan<T>(), gko::nan<T>()}, this->exec);
+
+    this->mtx1->apply(alpha, x, beta, y);
+
+    GKO_ASSERT_MTX_NEAR(y, l({-13.0, -5.0}), 0.0);
+}
+
+
 TYPED_TEST(Ell, MixedAppliesLinearCombinationToDenseVector1)
 {
     // Both vectors have the same value type which differs from the matrix

reference/test/matrix/sellp_kernels.cpp

@@ -1,4 +1,4 @@
-// SPDX-FileCopyrightText: 2017 - 2024 The Ginkgo authors
+// SPDX-FileCopyrightText: 2017 - 2025 The Ginkgo authors
 //
 // SPDX-License-Identifier: BSD-3-Clause
 
@@ -114,6 +114,21 @@ TYPED_TEST(Sellp, AppliesLinearCombinationToDenseVector)
 }
 
 
+TYPED_TEST(Sellp, AppliesLinearCombinationToDenseVectorWithZeroBetaNaN)
+{
+    using T = typename TestFixture::value_type;
+    using Vec = typename TestFixture::Vec;
+    auto alpha = gko::initialize<Vec>({-1.0}, this->exec);
+    auto beta = gko::initialize<Vec>({0.0}, this->exec);
+    auto x = gko::initialize<Vec>({2.0, 1.0, 4.0}, this->exec);
+    auto y = gko::initialize<Vec>({gko::nan<T>(), gko::nan<T>()}, this->exec);
+
+    this->mtx1->apply(alpha, x, beta, y);
+
+    GKO_ASSERT_MTX_NEAR(y, l({-13.0, -5.0}), 0.0);
+}
+
+
 TYPED_TEST(Sellp, AppliesLinearCombinationToMixedDenseVector)
 {
     using value_type = gko::next_precision<typename TestFixture::value_type>;

reference/test/matrix/sparsity_csr_kernels.cpp

@@ -1,4 +1,4 @@
-// SPDX-FileCopyrightText: 2017 - 2024 The Ginkgo authors
+// SPDX-FileCopyrightText: 2017 - 2025 The Ginkgo authors
 //
 // SPDX-License-Identifier: BSD-3-Clause
 
@@ -190,6 +190,22 @@ TYPED_TEST(SparsityCsr, AppliesLinearCombinationToDenseVector)
 }
 
 
+TYPED_TEST(SparsityCsr, AppliesLinearCombinationToDenseVectorWithZeroBetaNaN)
+{
+    using Vec = typename TestFixture::Vec;
+    using T = typename TestFixture::value_type;
+    auto alpha = gko::initialize<Vec>({-1.0}, this->exec);
+    auto beta = gko::initialize<Vec>({0.0}, this->exec);
+    auto x = gko::initialize<Vec>({2.0, 1.0, 4.0}, this->exec);
+    auto y = gko::initialize<Vec>({gko::nan<T>(), gko::nan<T>()}, this->exec);
+
+    this->mtx->apply(alpha, x, beta, y);
+
+    EXPECT_EQ(y->at(0), T{-7.0});
+    EXPECT_EQ(y->at(1), T{-1.0});
+}
+
+
 TYPED_TEST(SparsityCsr, AppliesLinearCombinationToMixedDenseVector)
 {
     using T = gko::next_precision<typename TestFixture::value_type>;