From c1fb92a51674312595316dc225cccb7d5f000fc0 Mon Sep 17 00:00:00 2001
From: Junwha <qbit@unist.ac.kr>
Date: Mon, 1 Apr 2024 04:02:39 +0900
Subject: [PATCH] [Bugfix] Use correct num_triplets

---
 taichi/program/sparse_matrix.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/taichi/program/sparse_matrix.cpp b/taichi/program/sparse_matrix.cpp
index 2432b56ba886a..819572b80a356 100644
--- a/taichi/program/sparse_matrix.cpp
+++ b/taichi/program/sparse_matrix.cpp
@@ -380,7 +380,7 @@ void make_sparse_matrix_from_ndarray(Program *prog,
                                      const Ndarray &ndarray) {
   std::string sdtype = taichi::lang::data_type_name(sm.get_data_type());
   auto data_ptr = prog->get_ndarray_data_ptr_as_int(&ndarray);
-  auto num_triplets = ndarray.get_nelement() * ndarray.get_element_size() / 3;
+  auto num_triplets = ndarray.get_nelement();
   if (sdtype == "f32") {
     build_ndarray_template<float32>(sm, data_ptr, num_triplets);
   } else if (sdtype == "f64") {