Skip to content

Commit b978d41

Browse files
committed
Cast dims to Float64 in contract sequence finding
1 parent 0b2426c commit b978d41

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ext/ITensorNetworksTensorOperationsExt/ITensorNetworksTensorOperationsExt.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ using TensorOperations: TensorOperations, optimaltree
77

88
function ITensorNetworks.contraction_sequence(::Algorithm"optimal", tn::ITensorList)
99
network = collect.(inds.(tn))
10-
inds_to_dims = Dict(i => dim(i) for i in unique(reduce(vcat, network)))
10+
inds_to_dims = Dict(i => Float64(dim(i)) for i in unique(reduce(vcat, network)))
1111
seq, _ = optimaltree(network, inds_to_dims)
1212
return seq
1313
end

0 commit comments

Comments
 (0)