Fix extensions on julia 1.6

Jutho · May 29, 2024 · 02af560 · 02af560
1 parent f82dae0
commit 02af560
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/ext/TensorKitChainRulesCoreExt.jl b/ext/TensorKitChainRulesCoreExt.jl
@@ -11,7 +11,11 @@ import TensorOperations as TO
 using TensorOperations: Backend, promote_contract
 using VectorInterface: promote_scale, promote_add
 
-ext = Base.get_extension(TensorOperations, :TensorOperationsChainRulesCoreExt)
+ext = @static if isdefined(Base, :get_extension)
+    Base.get_extension(TensorOperations, :TensorOperationsChainRulesCoreExt)
+else
+    TensorOperations.TensorOperationsChainRulesCoreExt
+end
 const _conj = ext._conj
 const trivtuple = ext.trivtuple
 

diff --git a/test/ad.jl b/test/ad.jl
@@ -4,7 +4,11 @@ using Random
 using FiniteDifferences
 using LinearAlgebra
 
-_repartition = Base.get_extension(TensorKit, :TensorKitChainRulesCoreExt)._repartition
+const _repartition = @static if isdefined(Base, :get_extension)
+    Base.get_extension(TensorKit, :TensorKitChainRulesCoreExt)._repartition
+else
+    TensorKit.TensorKitChainRulesCoreExt._repartition
+end
 
 # Test utility
 # -------------