init (apache#7943)

neo-ai · May 11, 2021 · 7c177f3 · 7c177f3
1 parent 30edeeb
commit 7c177f3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/te/autodiff/ad_simplify.cc b/src/te/autodiff/ad_simplify.cc
@@ -545,7 +545,7 @@ class EliminateDivModMutator : public ExprMutator {
       }
     }
 
-    return truncdiv(VisitExpr(op->a), VisitExpr(op->b));
+    return div(VisitExpr(op->a), VisitExpr(op->b));
   }
 
   virtual PrimExpr VisitExpr_(const ModNode* op) {