Updates and small fixes

Author: kshyatt

Project.toml

@@ -45,7 +45,7 @@ CUDA = "5.9"
 ChainRulesCore = "1"
 ChainRulesTestUtils = "1"
 Combinatorics = "1"
-Enzyme = "0.13.131"
+Enzyme = "0.13.134"
 EnzymeTestUtils = "0.2.5"
 FiniteDifferences = "0.12"
 GPUArrays = "11.3.1"
@@ -95,6 +95,3 @@ cuTENSOR = "011b41b2-24ef-40a8-b3eb-fa098493e9e1"
 
 [targets]
 test = ["ArgParse", "Adapt", "Aqua", "AllocCheck", "Combinatorics", "CUDA", "cuTENSOR", "GPUArrays", "LinearAlgebra", "SafeTestsets", "TensorOperations", "Test", "TestExtras", "ChainRulesCore", "ChainRulesTestUtils", "FiniteDifferences", "Zygote", "Mooncake", "Enzyme", "EnzymeTestUtils", "JET"]
-
-[sources]
-TensorOperations = {url = "https://github.com/quantumkithub/tensoroperations.jl", rev = "ksh/enzyme_update"}

ext/TensorKitEnzymeExt/linalg.jl

@@ -159,7 +159,7 @@ function EnzymeRules.augmented_primal(
     ) where {RT}
     ret = inv(A.val)
     primal = EnzymeRules.needs_primal(config) ? ret : nothing
-    shadow = EnzymeRules.needs_shadow(config) ? zero(ret) : nothing
+    shadow = EnzymeRules.needs_shadow(config) ? make_zero(ret) : nothing
     cache = (ret, shadow)
     return EnzymeRules.AugmentedReturn(primal, shadow, cache)
 end

ext/TensorKitEnzymeExt/utility.jl

@@ -56,8 +56,9 @@ end
 
 @inline EnzymeRules.inactive_type(::Type{<:TensorKit.FusionTree}) = true
 @inline EnzymeRules.inactive_type(::Type{<:TensorKit.GenericTreeTransformer}) = true
+@inline EnzymeRules.inactive_type(::Type{<:TensorKit.VectorSpace}) = true
 
-@inline EnzymeRules.inactive(::typeof(TensorKit.fusionblockstructure), arg) = nothing
+@inline EnzymeRules.inactive(::typeof(TensorKit.fusionblockstructure), arg::Any) = nothing
 @inline EnzymeRules.inactive(::typeof(TensorKit.select), s::HomSpace, i::Index2Tuple) = nothing
 @inline EnzymeRules.inactive(::typeof(TensorKit.flip), s::HomSpace, i::Any) = nothing
 @inline EnzymeRules.inactive(::typeof(TensorKit.permute), s::HomSpace, i::Index2Tuple) = nothing