Fixed MGPCG (#652)

taichi-dev · Mar 25, 2020 · 6fcba96 · 6fcba96
1 parent 76ff830
commit 6fcba96
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/examples/mgpcg_advanced.py b/examples/mgpcg_advanced.py
@@ -54,9 +54,9 @@ def init(self):
                 ti.ndrange(*(
                     (self.N_ext, self.N_tot - self.N_ext), ) * self.dim)):
             self.r[0][I] = 1.0
-            for i in ti.static(range(self.dim)):
-                self.r[0][I] *= ti.sin(2.0 * np.pi * (i - self.N_ext) * 2.0 /
-                                       self.N_tot)
+            for k in ti.static(range(self.dim)):
+                self.r[0][I] *= ti.sin(2.0 * np.pi *
+                                       (I[k] - self.N_ext) * 2.0 / self.N_tot)
             self.z[0][I] = 0.0
             self.Ap[I] = 0.0
             self.p[I] = 0.0