sparsity detection from Symbolics

Author: ArnoStrouwen

Project.toml

@@ -16,7 +16,6 @@ LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 Requires = "ae029012-a4dd-5104-9daa-d747884805df"
 SparseArrays = "2f01184e-e22b-5df5-ae63-d93ebab69eaf"
 StaticArrays = "90137ffa-7385-5640-81b9-e52037218182"
-Symbolics = "0c5d862f-8b57-4792-8d23-62f2024744c7"
 VertexSafeGraphs = "19fa3120-7c27-5ec5-8db8-b0b0aa330d6f"
 
 [compat]
@@ -43,8 +42,9 @@ Pkg = "44cfe95a-1eb2-52ea-b672-e2afdf69b78f"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 SafeTestsets = "1bc83da4-3b8d-516f-aca4-4fe02f6d838f"
 StaticArrays = "90137ffa-7385-5640-81b9-e52037218182"
+Symbolics = "0c5d862f-8b57-4792-8d23-62f2024744c7"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 
 [targets]
-test = ["Test", "ArrayInterfaceBandedMatrices", "ArrayInterfaceBlockBandedMatrices", "BandedMatrices", "BlockBandedMatrices", "IterativeSolvers", "Pkg", "Random", "SafeTestsets", "Zygote", "StaticArrays"]
+test = ["Test", "ArrayInterfaceBandedMatrices", "ArrayInterfaceBlockBandedMatrices", "BandedMatrices", "BlockBandedMatrices", "IterativeSolvers", "Pkg", "Random", "SafeTestsets", "Symbolics", "Zygote", "StaticArrays"]

src/SparseDiffTools.jl

@@ -11,7 +11,6 @@ using Adapt
 
 using LinearAlgebra
 using SparseArrays, ArrayInterfaceCore
-using Symbolics
 
 import StaticArrays, ArrayInterfaceStaticArrays
 
@@ -60,7 +59,6 @@ include("differentiation/compute_jacobian_ad.jl")
 include("differentiation/compute_hessian_ad.jl")
 include("differentiation/jaches_products.jl")
 include("differentiation/vecjac_products.jl")
-include("differentiation/sparsity_detection.jl")
 
 Base.@pure __parameterless_type(T) = Base.typename(T).wrapper
 parameterless_type(x) = parameterless_type(typeof(x))

src/differentiation/sparsity_detection.jl

@@ -1,4 +1,5 @@
 using SparseDiffTools
+using Symbolics
 using FiniteDiff: finite_difference_jacobian, finite_difference_jacobian!
 
 using LinearAlgebra, SparseArrays, Test
@@ -36,7 +37,7 @@ function second_derivative_stencil(N)
 end
 
 output = ones(30); input = ones(30)
-sparsity_pattern = jacobian_sparsity(f,output,input)
+sparsity_pattern = Symbolics.jacobian_sparsity(f,output,input)
 true_jac = Float64.(sparse(sparsity_pattern))
 colors = matrix_colors(true_jac)
 @test colors == repeat(1:3,10)

test/test_integration.jl

@@ -1,5 +1,6 @@
 ## Hessian tests
 using SparseDiffTools
+using Symbolics
 using ForwardDiff
 using LinearAlgebra, SparseArrays
 
@@ -8,7 +9,7 @@ function fscalar(x)
 end
 
 x = randn(5)
-sparsity = hessian_sparsity(fscalar, x)
+sparsity = Symbolics.hessian_sparsity(fscalar, x)
 colors = matrix_colors(tril(sparsity))
 ncolors = maximum(colors)
 D = hcat([float.(i .== colors) for i in 1:ncolors]...)