Skip to content

Commit ed6de5a

Browse files
authored
Add COO to CSR test (#921)
1 parent 7b7d863 commit ed6de5a

File tree

1 file changed

+17
-1
lines changed

1 file changed

+17
-1
lines changed

test/00_sparse/Convert.cu

+17-1
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,7 @@ TYPED_TEST(ConvertSparseTestsAll, ConvertCOO) {
8080
const auto n = D.Size(1);
8181

8282
// Convert dense D to sparse S.
83-
auto S = experimental::make_zero_tensor_coo<TestType, index_t>({m, n});
83+
auto S = experimental::make_zero_tensor_coo<TestType, int>({m, n});
8484
(S = dense2sparse(D)).run(exec);
8585
ASSERT_EQ(S.Rank(), 2);
8686
ASSERT_EQ(S.Size(0), m);
@@ -111,6 +111,22 @@ TYPED_TEST(ConvertSparseTestsAll, ConvertCOO) {
111111
}
112112
}
113113

114+
// Convert sparse S to another sparse format.
115+
auto Acsr = experimental::make_zero_tensor_csr<TestType, int, int>({m, n});
116+
(Acsr = sparse2sparse(S)).run(exec);
117+
ASSERT_EQ(Acsr.Rank(), 2);
118+
ASSERT_EQ(Acsr.Size(0), m);
119+
ASSERT_EQ(Acsr.Size(1), n);
120+
ASSERT_EQ(Acsr.Nse(), 4);
121+
122+
// Getters are expensive, but fully functional!
123+
exec.sync();
124+
for (index_t i = 0; i < m; i++) {
125+
for (index_t j = 0; j < n; j++) {
126+
ASSERT_EQ(Acsr(i, j), D(i, j));
127+
}
128+
}
129+
114130
MATX_EXIT_HANDLER();
115131
}
116132

0 commit comments

Comments
 (0)