Specialize on operations

Author: charleskawczynski

Project.toml

@@ -1,7 +1,7 @@
 name = "NullBroadcasts"
 uuid = "0d71be07-595a-4f89-9529-4065a4ab43a6"
 authors = ["CliMA Contributors <[email protected]>"]
-version = "0.1.0"
+version = "0.1.1"
 
 [compat]
 Aqua = "0.8"

src/NullBroadcasts.jl

@@ -59,6 +59,11 @@ Base.broadcasted(op::typeof(+), arg, ::NullBroadcasted, args...) =
 Base.broadcasted(op::typeof(+), a::NullBroadcasted, ::NullBroadcasted, args...) =
     Base.broadcasted(op, a, args...)
 
+# Specialize on identity cases:
+Base.broadcasted(::typeof(+), ::NullBroadcasted, x) = x
+Base.broadcasted(::typeof(+), x, ::NullBroadcasted) = x
+Base.broadcasted(::typeof(-), x, ::NullBroadcasted) = x
+
 Base.broadcasted(op::typeof(*), ::NullBroadcasted, args...) = NullBroadcasted()
 Base.broadcasted(op::typeof(*), arg, ::NullBroadcasted) = NullBroadcasted()
 Base.broadcasted(op::typeof(*), ::NullBroadcasted, ::NullBroadcasted) = NullBroadcasted()