Add sparse_storage for Adjoint and Transpose

ITensor · Nov 15, 2024 · d1a5b53 · d1a5b53
1 parent 3efdb89
commit d1a5b53
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/NDTensors/src/lib/SparseArrayInterface/src/abstractsparsearray/wrappedabstractsparsearray.jl b/NDTensors/src/lib/SparseArrayInterface/src/abstractsparsearray/wrappedabstractsparsearray.jl
@@ -13,8 +13,10 @@ function stored_indices(a::Adjoint)
   return Iterators.map(I -> CartesianIndex(reverse(Tuple(I))), stored_indices(parent(a)))
 end
 stored_length(a::Adjoint) = stored_length(parent(a))
+sparse_storage(a::Adjoint) = Iterators.map(adjoint, sparse_storage(parent(a)))
 
 function stored_indices(a::Transpose)
   return Iterators.map(I -> CartesianIndex(reverse(Tuple(I))), stored_indices(parent(a)))
 end
 stored_length(a::Transpose) = stored_length(parent(a))
+sparse_storage(a::Transpose) = Iterators.map(transpose, sparse_storage(parent(a)))