Merge pull request #394 from andrewrosemberg/ar/sortperm

Add Rules for sortperm and friends

Author: mzgubic

Project.toml

@@ -1,6 +1,6 @@
 name = "ChainRules"
 uuid = "082447d4-558c-5d27-93f4-14fc19e9eca2"
-version = "0.7.56"
+version = "0.7.57"
 
 [deps]
 ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"

src/rulesets/Base/nondiff.jl

@@ -289,6 +289,8 @@ VERSION >= v"1.3" && @non_differentiable istaskfailed(::Task)
 VERSION >= v"1.4" && @non_differentiable only(::Char)
 @non_differentiable open(::Any)
 
+@non_differentiable partialsortperm(::AbstractVector, ::Union{Integer, OrdinalRange})
+@non_differentiable partialsortperm!(::AbstractVector{<:Integer}, ::AbstractVector, ::Union{Integer, OrdinalRange})
 @non_differentiable pointer(::Any)
 @non_differentiable popdisplay(::AbstractDisplay)
 @non_differentiable position(::Base.Filesystem.File)
@@ -350,6 +352,8 @@ end
 @non_differentiable size(::Any, ::Any)
 @non_differentiable sizeof(::Any)
 @non_differentiable sleep(::Any)
+@non_differentiable sortperm(::AbstractVector)
+@non_differentiable sortperm!(::AbstractVector{<:Integer}, ::AbstractVector)
 @non_differentiable split(::AbstractString)
 @non_differentiable split(::AbstractString, ::AbstractChar)
 @non_differentiable splitdir(::AbstractString)