Fix performance of MA.add_mul for NonlinearExpr (#3475)

jump-dev · Sep 1, 2023 · 6553341 · 6553341
1 parent 10a7ce5
commit 6553341
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/nlp_expr.jl b/src/nlp_expr.jl
@@ -370,14 +370,14 @@ end
 function _MA.operate!!(
     ::typeof(_MA.add_mul),
     x::GenericNonlinearExpr,
-    y::AbstractJuMPScalar,
-)
+    args::Vararg{Any,N},
+) where {N}
     _throw_if_not_real(x)
     if x.head == :+
-        push!(x.args, y)
+        push!(x.args, *(args...))
         return x
     end
-    return +(x, y)
+    return +(x, *(args...))
 end
 
 """