Fix more tests

ITensor · Oct 24, 2023 · 42bd16c · 42bd16c
1 parent d9a5520
commit 42bd16c
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/NDTensors/src/diag/diagtensor.jl b/NDTensors/src/diag/diagtensor.jl
@@ -125,6 +125,14 @@ function dense(::Type{<:AbstractArray}, T::DiagTensor)
   return adapt(leaf_parenttype(T), D_cpu)
 end
 
+# UniformDiag version
+# TODO: Delete once new DiagonalArray is designed.
+# TODO: This creates a tensor on CPU by default so may cause
+# problems for GPU.
+function dense(::Type{<:Number}, T::DiagTensor)
+  return dense(Tensor(Diag(fill(getdiagindex(T, 1), diaglength(T))), inds(T)))
+end
+
 denseblocks(T::DiagTensor) = dense(T)
 
 function permutedims!(