Fix setting the LinearSolve library at extension load

.github/workflows/ci.yml

@@ -18,7 +18,7 @@ jobs:
     strategy:
       matrix:
         version:
-          - 'min'
+          - '1.10'
           - '1.11' # to test inference
           - '1'
       fail-fast: false

Project.toml

@@ -1,6 +1,6 @@
 name = "SymbolicCompilerPasses"
 uuid = "3384d301-0fbe-4b40-9ae0-b0e68bedb069"
-version = "0.1.1"
+version = "0.1.2"
 authors = ["Dhairya Gandhi <dhairya@juliahub.com>"]
 
 [deps]
@@ -12,18 +12,14 @@ SymbolicUtils = "d1185830-fcd6-423d-90d6-eec64667417b"
 [weakdeps]
 LinearSolve = "7ed4a6bd-45f5-4d41-b270-4a48e9bafcae"
 
-[sources.SymbolicUtils]
-rev = "dg/opt_api"
-url = "https://github.com/DhairyaLGandhi/SymbolicUtils.jl"
-
 [extensions]
 SCPLinearSolveExt = ["LinearSolve"]
 
 [compat]
 LinearAlgebra = "1"
 LinearSolve = "3.53.0"
 PreallocationTools = "0.4.34, 1"
-StaticArrays = "1.9.15"
+StaticArrays = "1.9.15, 1"
 SymbolicUtils = "4.1.0"
 julia = "1"
 

ext/SCPLinearSolveExt/SCPLinearSolveExt.jl

@@ -4,9 +4,9 @@ using SymbolicUtils
 using SymbolicUtils.Code
 using LinearSolve
 using LinearAlgebra
-import SymbolicCompilerPasses: ldiv_transformation, SymbolicCompilerPasses, get_factorization, get_from_cache, FACTORIZATION_CACHE
+import SymbolicCompilerPasses: ldiv_transformation, SymbolicCompilerPasses, get_factorization, get_from_cache, FACTORIZATION_CACHE, LINEARSOLVE_LIB
 
-SymbolicCompilerPasses.LINEARSOLVE_LIB[] = true
+__init__() = SymbolicCompilerPasses.LINEARSOLVE_LIB[] = true
 
 function linear_solve(A, B)
     linsolve = get_factorization(A, B)

src/hvncat_static_opt.jl

@@ -251,7 +251,7 @@ const HVNCAT_STATIC_RULE = OptimizationRule(
 
 function literal_static_opt(expr, state::CSEState)
     # Try to apply optimization rules
-    optimized = apply_optimization_rules(expr, state, HVNCAT_STATIC_RULE)
+    optimized = apply_optimization_rules(expr, state, [HVNCAT_STATIC_RULE])
     if optimized !== nothing
         return optimized
     end

src/la_opt.jl

@@ -77,7 +77,7 @@ const NORMALIZE_RULE = GenericRule("normalize", LinearAlgebra.normalize, LinearA
 
 function triu_opt(expr, state)
     # Try to apply optimization rules
-    optimized = apply_optimization_rules(expr, state, TRIU_RULE)
+    optimized = apply_optimization_rules(expr, state, [TRIU_RULE])
     if optimized !== nothing
         return optimized
     end

src/ldiv_opt.jl

@@ -230,7 +230,7 @@ const LDIV_RULE = OptimizationRule(
 function ldiv_opt(expr, state::CSEState)
 
     # Try to apply optimization rules
-    optimized = apply_optimization_rules(expr, state, LDIV_RULE)
+    optimized = apply_optimization_rules(expr, state, [LDIV_RULE])
     if optimized !== nothing
         return optimized
     end

src/matmuladd.jl

@@ -176,7 +176,7 @@ const MATMUL_ADD_RULE = OptimizationRule(
 function mul5_opt(expr, state::CSEState)
 
     # Try to apply optimization rules
-    optimized = apply_optimization_rules(expr, state, MATMUL_ADD_RULE)
+    optimized = apply_optimization_rules(expr, state, [MATMUL_ADD_RULE])
     if optimized !== nothing
         return optimized
     end