Merge pull request #911 from SciML/functionproperties

Move hasbranching to FunctionProperties.jl

Project.toml

@@ -7,17 +7,16 @@ version = "7.41.0"
 ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b"
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
 ArrayInterface = "4fba245c-0d91-5ea0-9b3e-6abc04ee57a9"
-Cassette = "7057c7e9-c182-5462-911a-8362d720325c"
 ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
 DiffEqBase = "2b5f629d-d688-5b77-993f-72d75c75574e"
 DiffEqCallbacks = "459566f4-90b8-5000-8ac3-15dfb0a30def"
 DiffEqNoiseProcess = "77a26b50-5914-5dd7-bc55-306e6241c503"
-DiffRules = "b552c78f-8df3-52c6-915a-8e097449b14b"
 Distributions = "31c24e10-a181-5473-b8eb-7969acd0382f"
 EllipsisNotation = "da5c29d0-fa7d-589e-88eb-ea29b0a81949"
 Enzyme = "7da242da-08ed-463a-9acd-ee780be4f1d9"
 FiniteDiff = "6a86dc24-6348-571c-b903-95158fe2bd41"
 ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210"
+FunctionProperties = "f62d2435-5019-4c03-9749-2d4c77af0cbc"
 FunctionWrappersWrappers = "77dc65aa-8811-40c2-897b-53d922fa7daf"
 GPUArraysCore = "46192b85-c4d5-4398-a991-12ede77f4527"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
@@ -48,17 +47,16 @@ ZygoteRules = "700de1a5-db45-46bc-99cf-38207098b444"
 ADTypes = "0.1, 0.2"
 Adapt = "1.0, 2.0, 3.0"
 ArrayInterface = "7"
-Cassette = "0.3.6"
 ChainRulesCore = "0.10.7, 1"
 DiffEqBase = "6.93"
 DiffEqCallbacks = "2.29"
 DiffEqNoiseProcess = "4.1.4, 5.0"
-DiffRules = "1"
 Distributions = "0.24, 0.25"
 EllipsisNotation = "1"
 Enzyme = "0.11.6"
 FiniteDiff = "2"
 ForwardDiff = "0.10"
+FunctionProperties = "0.1"
 FunctionWrappersWrappers = "0.1"
 GPUArraysCore = "0.1"
 LinearSolve = "2"

src/SciMLSensitivity.jl

@@ -21,10 +21,8 @@ import TruncatedStacktraces
 
 import PreallocationTools: dualcache, get_tmp, DiffCache, LazyBufferCache
 import FunctionWrappersWrappers
-
-using Cassette, DiffRules
-using Core: CodeInfo, SlotNumber, SSAValue, ReturnNode, GotoIfNot
 using EllipsisNotation
+using FunctionProperties: hasbranching
 
 using Markdown
 
@@ -41,7 +39,6 @@ import SciMLBase: AbstractOverloadingSensitivityAlgorithm, AbstractSensitivityAl
     AbstractSecondOrderSensitivityAlgorithm,
     AbstractShadowingSensitivityAlgorithm
 
-include("hasbranching.jl")
 include("sensitivity_algorithms.jl")
 include("derivative_wrappers.jl")
 include("sensitivity_interface.jl")

test/runtests.jl

@@ -53,9 +53,6 @@ end
     end
 
     if GROUP == "All" || GROUP == "Core2"
-        @time @safetestset "hasbranching" begin
-            include("hasbranching.jl")
-        end
         @time @safetestset "Literal Adjoint" begin
             include("literal_adjoint.jl")
         end