make sure that DistArray of (non-async) lazy tiles can be reduced

ValeevGroup · Oct 30, 2024 · 63d3201 · 63d3201
1 parent 486ae16
commit 63d3201
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/tests/expressions_mixed.cpp b/tests/expressions_mixed.cpp
@@ -176,6 +176,10 @@ BOOST_AUTO_TEST_CASE(mult_factories) {
   // BOOST_CHECK_NO_THROW(w("a,b") = u("a,b") * v("a,b"));
 }
 
+BOOST_AUTO_TEST_CASE(reduction_factories) {
+  BOOST_CHECK_NO_THROW(v("a,b").norm());
+}
+
 BOOST_AUTO_TEST_CASE(kronecker) {
 #if !MULT_DENSE_SPARSE_TO_SPARSE
   // ok

diff --git a/tests/sparse_tile.h b/tests/sparse_tile.h
@@ -638,7 +638,9 @@ typename EigenSparseTile<T, TagType>::numeric_type product(
 // foreach(i) result += arg[i] * arg[i]
 template <typename T, typename TagType>
 typename EigenSparseTile<T, TagType>::numeric_type squared_norm(
-    const EigenSparseTile<T, TagType>& arg);
+    const EigenSparseTile<T, TagType>& arg) {
+  return arg.matrix().squaredNorm();
+}
 // sqrt(squared_norm(arg))
 template <typename T, typename TagType>
 typename EigenSparseTile<T, TagType>::numeric_type norm(