fix treewidth solver dispatch issue

TensorBFS · Oct 19, 2024 · 1149b6d · 1149b6d
1 parent 3713c9b
commit 1149b6d
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "OMEinsumContractionOrders"
 uuid = "6f22d1fd-8eed-4bb7-9776-e7d684900715"
 authors = ["Jin-Guo Liu", "Pan Zhang"]
-version = "0.9.2"
+version = "0.9.3"
 
 [deps]
 AbstractTrees = "1520ce14-60c1-5f80-bbc7-55ef81b5835c"

diff --git a/src/kahypar.jl b/src/kahypar.jl
@@ -146,7 +146,7 @@ function optimize_kahypar(code::EinCode, size_dict; sc_target, max_group_size=40
     recursive_bipartite_optimize(bipartiter, code, size_dict)
 end
 
-function recursive_bipartite_optimize(bipartiter, code::EinCode, size_dict)
+function recursive_bipartite_optimize(bipartiter, code::AbstractEinsum, size_dict)
     ixs, iy = getixsv(code), getiyv(code)
     ixv = [ixs..., iy]
 

diff --git a/src/treewidth.jl b/src/treewidth.jl
@@ -135,7 +135,7 @@ end
 Optimizing the contraction order via solve the exact tree width of the line graph corresponding to the eincode and return a `NestedEinsum` object.
 Check the docstring of `exact_treewidth_method` for detailed explaination of other input arguments.
 """
-function optimize_exact_treewidth(optimizer::ExactTreewidth{GM}, code::EinCode{L}, size_dict::Dict) where {L,GM}
+function optimize_exact_treewidth(optimizer::ExactTreewidth{GM}, code::AbstractEinsum, size_dict::Dict) where {GM}
     optimize_exact_treewidth(optimizer, getixsv(code), getiyv(code), size_dict)
 end
 function optimize_exact_treewidth(optimizer::ExactTreewidth{GM}, ixs::AbstractVector{<:AbstractVector}, iy::AbstractVector, size_dict::Dict{L,TI}) where {L, TI, GM}