Only expand @. macros

SciML · Jul 28, 2017 · 8b05ba9 · 8b05ba9
1 parent 79b93ea
commit 8b05ba9
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/utils.jl b/src/utils.jl
@@ -38,9 +38,9 @@ end
 
 function to_muladd(ex::Expr)
     if !isaddition(ex)
-        if ex.head == :macrocall
-            # expand macros first (enables use of @. inside of @muladd expression)
-            return to_muladd(macroexpand(ex))
+        if ex.head == :macrocall && length(ex.args)==2 && ex.args[1] == Symbol("@__dot__")
+            # expand @. macros first (enables use of @. inside of @muladd expression)
+            return to_muladd(Base.Broadcast.__dot__(ex.args[2]))
         else
             # if expression is no sum apply the reduction to its arguments
             return Expr(ex.head, to_muladd.(ex.args)...)