Update compute_squared_norm2

ginkgo-project · Mar 7, 2023 · 973dadf · 973dadf
1 parent d815378
commit 973dadf
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/core/matrix/dense.cpp b/core/matrix/dense.cpp
@@ -498,12 +498,10 @@ void Dense<ValueType>::compute_squared_norm2(LinOp* result,
 template <typename ValueType>
 void Dense<ValueType>::compute_squared_norm2_impl(LinOp* result) const
 {
-    GKO_ASSERT_EQUAL_DIMENSIONS(result, dim<2>(1, this->get_size()[1]));
     auto exec = this->get_executor();
-    auto dense_res =
-        make_temporary_conversion<remove_complex<ValueType>>(result);
     array<char> tmp{exec};
-    exec->run(dense::make_compute_squared_norm2(this, dense_res.get(), tmp));
+    this->compute_squared_norm2(make_temporary_output_clone(exec, result).get(),
+                                tmp);
 }