diff --git a/Project.toml b/Project.toml index ff794c3122..5ac5f7bc0b 100644 --- a/Project.toml +++ b/Project.toml @@ -1,7 +1,7 @@ name = "ITensors" uuid = "9136182c-28ba-11e9-034c-db9fb085ebd5" authors = ["Matthew Fishman ", "Miles Stoudenmire "] -version = "0.7.11" +version = "0.7.12" [deps] Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e" diff --git a/ext/ITensorsChainRulesCoreExt/itensor.jl b/ext/ITensorsChainRulesCoreExt/itensor.jl index a830cc77bd..26d00baf18 100644 --- a/ext/ITensorsChainRulesCoreExt/itensor.jl +++ b/ext/ITensorsChainRulesCoreExt/itensor.jl @@ -27,7 +27,7 @@ function rrule(::Type{ITensor}, x1::AllowAlias, x2::TensorStorage, x3) y = ITensor(x1, x2, x3) function ITensor_pullback(ȳ) x̄1 = NoTangent() - x̄2 = ȳ.tensor.storage + x̄2 = unthunk(ȳ).tensor.storage x̄3 = NoTangent() return (NoTangent(), x̄1, x̄2, x̄3) end @@ -84,7 +84,7 @@ end function rrule(::typeof(tensor), x1::ITensor) y = tensor(x1) function tensor_pullback(ȳ) - x̄1 = ITensor(typeof(storage(x1))(ȳ.storage.data), inds(x1)) + x̄1 = ITensor(typeof(storage(x1))(unthunk(ȳ).storage.data), inds(x1)) return (NoTangent(), x̄1) end return y, tensor_pullback