chore: add a precompilation workload

Author: AayushSabharwal

Project.toml

@@ -20,6 +20,7 @@ Moshi = "2e0e35c7-a2e4-4343-998d-7ef72827ed2d"
 MultivariatePolynomials = "102ac46a-7ee4-5c85-9060-abc95bfdeaa3"
 MutableArithmetics = "d8a4904e-b15c-11e9-3269-09a3773c0cb0"
 NaNMath = "77ba4419-2d1f-58cd-9bb1-8ffee604a2e3"
+PrecompileTools = "aea7be01-6a6a-4083-8856-8a6e6704d82a"
 ReadOnlyArrays = "988b38a3-91fc-5605-94a2-ee2116b3bd83"
 Setfield = "efcf1570-3423-57d1-acb7-fd33fddbac46"
 SparseArrays = "2f01184e-e22b-5df5-ae63-d93ebab69eaf"
@@ -61,6 +62,7 @@ MultivariatePolynomials = "0.5.12"
 MutableArithmetics = "1.6.5"
 NaNMath = "0.3, 1.1.2"
 OhMyThreads = "0.7"
+PrecompileTools = "1.2.1"
 ReadOnlyArrays = "0.2.0"
 ReverseDiff = "1"
 RuntimeGeneratedFunctions = "0.5.13"

src/SymbolicUtils.jl

@@ -34,6 +34,7 @@ import DynamicPolynomials as DP
 import MutableArithmetics as MA
 import LinearAlgebra
 import SparseArrays: SparseMatrixCSC, findnz, sparse
+import PrecompileTools
 
 macro manually_scope(val, expr, is_forced = false)
     @assert Meta.isexpr(val, :call)
@@ -101,7 +102,9 @@ include("small_array.jl")
 export istree, operation, arguments, sorted_arguments, iscall, unwrap_const
 # Sym, Term,
 # Add, Mul and Pow
-include("types.jl")
+PrecompileTools.@recompile_invalidations begin
+    include("types.jl")
+end
 
 include("printing.jl")
 
@@ -151,4 +154,26 @@ include("substitute.jl")
 
 include("code.jl")
 
+PrecompileTools.@setup_workload begin
+    PrecompileTools.@compile_workload begin
+        @syms x y f(t) q[1:5]
+        x + y
+        x * y
+        x / y
+        x ^ y
+        x ^ 5
+        6 ^ x
+        x - y
+        -y
+        f(x)
+        (5x / 5)
+        show(devnull, x ^ 2 + y * x + y / 3x)
+        expand((x + y) ^ 2)
+        simplify(x ^ (1//2) + (sin(x) ^ 2 + cos(x) ^ 2) + 2(x + y) - x - y)
+        substitute(x + 2y + sin(x), Dict(x => y); fold = false)
+        substitute(x + 2y + sin(x), Dict(x => 1); fold = true)
+        q[1]
+    end
+end
+
 end # module