fix (#49)

TensorBFS · Dec 11, 2024 · 20ef83b · 20ef83b
1 parent 855d98a
commit 20ef83b
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/src/complexity.jl b/src/complexity.jl
@@ -63,7 +63,7 @@ function _timespacereadwrite_complexity(ei::NestedEinsum, log2_sizes::Dict{L,VT}
     end
     tc2, sc2, rw2 = _timespacereadwrite_complexity(getixsv(ei.eins), getiyv(ei.eins), log2_sizes)
     tc = log2sumexp2([tcs..., tc2])
-    sc = max(reduce(max, scs), sc2)
+    sc = max(reduce(max, scs, init=zero(VT)), sc2)
     rw = log2sumexp2([rws..., rw2])
     return tc, sc, rw
 end

diff --git a/test/Core.jl b/test/Core.jl
@@ -24,4 +24,12 @@ using Test
         new_eincode = OMEinsum.decorate(closed_code)
         @test eincode(tensors...) ≈ new_eincode(tensors...)
     end
-end
+end
+
+@testset "contraction complexity for empty tensor network" begin
+    code = OMEinsumContractionOrders.EinCode(Vector{Char}[], Char[])
+    @test contraction_complexity(code, Dict{Char, Int}()).sc == 0
+
+    code = OMEinsumContractionOrders.NestedEinsum(OMEinsumContractionOrders.NestedEinsum{Char}[], OMEinsumContractionOrders.EinCode(Vector{Char}[], Char[]))
+    @test contraction_complexity(code, Dict{Char, Int}()).sc == 0
+end