minor bug fixes

owensgroup · Jul 13, 2024 · 601e378 · 601e378
1 parent 6f43f5d
commit 601e378
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/include/rxmesh/matrix/sparse_matrix.cuh b/include/rxmesh/matrix/sparse_matrix.cuh
@@ -537,7 +537,7 @@ struct SparseMatrix
  rxmesh::DenseMatrix<T>& C_mat,
  cudaStream_t stream = 0)
  {
- assert(cols() == B_mat.cols());
+ assert(cols() == B_mat.rows());
  assert(rows() == C_mat.rows());
  assert(B_mat.cols() == C_mat.cols());
 

diff --git a/tests/RXMesh_test/test_attribute.cuh b/tests/RXMesh_test/test_attribute.cuh
@@ -130,10 +130,14 @@ TEST(Attribute, Reduce)
  ASSERT_EQ(cudaDeviceSynchronize(), cudaSuccess);
 
  uint32_t result = 0;
- rx.for_each_edge(rxmesh::HOST, [&](const rxmesh::EdgeHandle eh) {
- auto pl = eh.unpack();
- result = std::max(result, pl.first * pl.second);
- });
+ rx.for_each_edge(
+ rxmesh::HOST,
+ [&](const rxmesh::EdgeHandle eh) {
+ auto pl = eh.unpack();
+ result = std::max(result, pl.first * pl.second);
+ },
+ NULL,
+ false);
 
  EXPECT_EQ(output, result);
 }