From 2023b82abd45b82bd55dca41562d91d23451271e Mon Sep 17 00:00:00 2001 From: Logan Bishop-Van Horn Date: Mon, 18 Sep 2023 15:01:22 -0700 Subject: [PATCH] Update solve.py --- tdgl/solver/solve.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tdgl/solver/solve.py b/tdgl/solver/solve.py index 7454b38..2d4f01b 100644 --- a/tdgl/solver/solve.py +++ b/tdgl/solver/solve.py @@ -366,7 +366,9 @@ def update( supercurrent + normal_current, use_cupy=use_cupy ) if use_cupy: - get_A_induced_cupy[128, 1024]( + threads_per_block = 1024 + num_blocks = int(len(edges) / threads_per_block) + get_A_induced_cupy[num_blocks, threads_per_block]( J_site, areas, sites, edge_centers, new_A_induced ) else: