Update solve.py

loganbvh · Sep 18, 2023 · 2023b82 · 2023b82
1 parent 4a4c9de
commit 2023b82
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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: