[example] Fix jacobian example (taichi-dev#6849)

Issue: taichi-dev#6733 The the grad field of variable at the left side of the global store will be cleared after taichi-dev#6579. Change to atomic add sort the issue.
quadpixels · May 13, 2023 · 6f4dda1 · 6f4dda1
1 parent 44cfcd4
commit 6f4dda1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/taichi/examples/autodiff/jacobian.py b/python/taichi/examples/autodiff/jacobian.py
@@ -15,7 +15,7 @@
 def func():
     for j in range(N_loss):
         for i in range(N_param):
-            losses[j] = ti.sin(j / 16 * np.pi) * ti.sin(
+            losses[j] += ti.sin(j / 16 * np.pi) * ti.sin(
                 i / 16 * np.pi) * params[i]