Merge pull request #637 from JuliaDiff/mz/nograd

add remaining `@nograd` from Zygote

Author: mzgubic

Project.toml

@@ -1,10 +1,11 @@
 name = "ChainRules"
 uuid = "082447d4-558c-5d27-93f4-14fc19e9eca2"
-version = "1.36.2"
+version = "1.37.0"
 
 [deps]
 ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
 Compat = "34da2185-b29b-5c13-b0c7-acf172513d20"
+Distributed = "8ba89e20-285c-5b6f-9357-94700520ee1b"
 IrrationalConstants = "92d709cd-6900-40b7-9082-c6be49f344b6"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"

src/ChainRules.jl

@@ -3,6 +3,7 @@ module ChainRules
 using Base.Broadcast: materialize, materialize!, broadcasted, Broadcasted, broadcastable
 using ChainRulesCore
 using Compat
+using Distributed
 using IrrationalConstants: logtwo, logten
 using LinearAlgebra
 using LinearAlgebra.BLAS
@@ -34,8 +35,11 @@ include("rulesets/Base/indexing.jl")
 include("rulesets/Base/sort.jl")
 include("rulesets/Base/mapreduce.jl")
 
+include("rulesets/Distributed/nondiff.jl")
+
 include("rulesets/Statistics/statistics.jl")
 
+include("rulesets/LinearAlgebra/nondiff.jl")
 include("rulesets/LinearAlgebra/utils.jl")
 include("rulesets/LinearAlgebra/blas.jl")
 include("rulesets/LinearAlgebra/lapack.jl")

src/rulesets/Base/nondiff.jl

@@ -93,6 +93,8 @@
 ##### Exported functions, alphabetically
 #####
 
+@non_differentiable (:)(::Any...) # same as the Colon() singleton instance
+
 @non_differentiable abspath(::AbstractString...)
 @non_differentiable all(::Any)
 @non_differentiable all(::Any, ::Any)
@@ -110,8 +112,8 @@
 @non_differentiable bytesavailable(::Any)
 
 @non_differentiable CartesianIndices(::Any)
+@non_differentiable Channel(::Any...)
 @non_differentiable cd(::AbstractString)
-@non_differentiable ceil(::Any...)
 @non_differentiable chomp(::AbstractString)
 @non_differentiable chop(::AbstractString)
 @non_differentiable cld(::Any, ::Any)
@@ -185,7 +187,6 @@
 @non_differentiable fld(::Any, ::Any, ::RoundingMode)
 @non_differentiable floatmax(::Any)
 @non_differentiable floatmin(::Any)
-@non_differentiable floor(::Any...)
 @non_differentiable flush(::Any)
 
 @non_differentiable gensym(::Symbol)
@@ -380,7 +381,6 @@ end
 @non_differentiable reset(::IO)
 @non_differentiable reverse(::AbstractString)
 @non_differentiable rm(::AbstractString)
-@non_differentiable round(::Any...)
 @non_differentiable rsplit(::AbstractString)
 @non_differentiable rsplit(::AbstractString, ::AbstractChar)
 @non_differentiable rstrip(::AbstractString)
@@ -470,6 +470,7 @@ elseif isdefined(Base, :cumulative_compile_time_ns)
     @non_differentiable Base.cumulative_compile_time_ns()
 end
 @non_differentiable Base.time_print(::Any...)
+@non_differentiable Base.OneTo(::Any...)
 
 @non_differentiable Broadcast.combine_styles(::Any...)
 @non_differentiable Broadcast.result_style(::Any)

src/rulesets/Distributed/nondiff.jl

@@ -0,0 +1,3 @@
+@non_differentiable Distributed.CachingPool(::Any...)
+@non_differentiable Distributed.WorkerPool(::Any...)
+@non_differentiable workers()

src/rulesets/LinearAlgebra/nondiff.jl

@@ -0,0 +1 @@
+@non_differentiable isposdef(::Any)