add rules for compilerbarrier

Author: oxinabox

Project.toml

@@ -1,6 +1,6 @@
 name = "ChainRules"
 uuid = "082447d4-558c-5d27-93f4-14fc19e9eca2"
-version = "1.48.1"
+version = "1.49.0"
 
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"

src/rulesets/Core/core.jl

@@ -32,3 +32,14 @@ function rrule(::typeof(ifelse), c, a::Number, b::Number)
     ifelse_pullback(Δ) = (NoTangent(), NoTangent(), ifelse(c, Δ, zero(Δ)), ifelse(c, zero(Δ), Δ))
     return ifelse(c, a, b), ifelse_pullback
 end
+
+if isdefined(Core, :compilerbarrier)
+    function ChainRulesCore.frule((_, _, Δ), ::typeof(Core.compilerbarrier), setting, value)
+        return (Core.compilerbarrier(setting, value), Δ)
+    end
+
+    function ChainRulesCore.rrule(::typeof(Core.compilerbarrier), setting, value)
+        compilerbarrier_pullback(Δ) = NoTangent(), NoTangent(), Δ
+        return (Core.compilerbarrier(setting, value), compilerbarrier_pullback)
+    end
+end

test/rulesets/Core/core.jl

@@ -10,3 +10,9 @@ end
     test_rrule(ifelse, true, [1.1], [2.0]; check_inferred=false)
     test_frule(ifelse, false, [1.1], [2.0]; check_inferred=false)
 end
+
+isdefined(Core, :compilerbarrier) && @testset "compilerbarrier" begin
+    test_frule(Core.compilerbarrier, :const, 42.0)
+    test_rrule(Core.compilerbarrier, :const, 42.0) 
+end
+