Update operators.py

loganbvh · Sep 20, 2023 · d71d366 · d71d366
1 parent 3abb3f7
commit d71d366
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tdgl/finite_volume/operators.py b/tdgl/finite_volume/operators.py
@@ -60,7 +60,7 @@ def _spmatrix_set_many(spmatrix, i, j, x):
         offsets, (i, j, M, N) = _get_spmatrix_offsets_cupy(spmatrix, i, j)
 
     mask = offsets > -1
-    spmatrix.data[mask] = x[mask]
+    spmatrix.data[offsets[mask]] = x[mask]
     if not mask.all():
         # only insertions remain
         mask = ~mask