Refactor chunked matrix for accommodating memory-mapped utility module (

#192) * Refactor inference chunked matrix * Using set to count chunk's nonzero rows
amzn · Dec 10, 2022 · aad94c6 · aad94c6
1 parent 8f648b9
commit aad94c6
Show file tree

Hide file tree

Showing 2 changed files with 218 additions and 157 deletions.
diff --git a/pecos/core/utils/matrix.hpp b/pecos/core/utils/matrix.hpp
@@ -253,7 +253,7 @@ namespace pecos {
             csr_t res;
             res.allocate(rows, cols, nnz);
             std::memcpy(res.col_idx, col_idx, sizeof(index_type) * nnz);
-            std::memcpy(res.val, val, sizeof(float) * nnz);
+            std::memcpy(res.val, val, sizeof(value_type) * nnz);
             std::memcpy(res.row_ptr, row_ptr, sizeof(mem_index_type) * (rows + 1));
             return res;
         }
@@ -386,7 +386,7 @@ namespace pecos {
             csc_t res;
             res.allocate(rows, cols, nnz);
             std::memcpy(res.row_idx, row_idx, sizeof(index_type) * nnz);
-            std::memcpy(res.val, val, sizeof(float) * nnz);
+            std::memcpy(res.val, val, sizeof(value_type) * nnz);
             std::memcpy(res.col_ptr, col_ptr, sizeof(mem_index_type) * (cols + 1));
             return res;
         }