def fill(A: ti.sparse_matrix_builder(), b: ti.sparse_matrix_builder(),
interval: ti.i32):
for i in range(n):
if i > 0:
A[i - 1, i] += -1.0
A[i, i] += 1
if i < n - 1:
A[i + 1, i] += -1.0
A[i, i] += 1.0
if i % interval == 0:
b[i, 0] += 1.0
def fill(Abuilder: ti.sparse_matrix_builder()):
for i, j in ti.ndrange(n, n):
Abuilder[i, j] += i + j
def fill(Abuilder: ti.sparse_matrix_builder()):
for i in range(n):
Abuilder[i, i] += i
def fill(Abuilder: ti.sparse_matrix_builder(),
Bbuilder: ti.sparse_matrix_builder()):
for i, j in ti.ndrange(n, k):
Abuilder[i, j] += i + j
for i, j in ti.ndrange(k, m):
Bbuilder[i, j] -= i + j
