Fix div_p jvp rule

0x45f · May 7, 2022 · dd98410 · dd98410
1 parent 8b2e5ee
commit dd98410
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/paddle/incubate/autograd/primrules.py b/python/paddle/incubate/autograd/primrules.py
@@ -410,7 +410,7 @@ def div_jvp(op, x_dot, y_dot):
     if y_dot is None:
         return div(x_dot, y)
     elif x_dot is None:
-        return div(mul(x, y_dot), mul(y, y))
+        return neg(div(mul(x, y_dot), mul(y, y)))
     else:
         t1 = div(x_dot, y)
         t2 = div(mul(x, y_dot), mul(y, y))