Fix KaHyPar tests with EinExprs

ITensor · Feb 13, 2024 · d965b14 · d965b14
1 parent bd132eb
commit d965b14
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/test/test_contraction_sequence.jl b/test/test_contraction_sequence.jl
@@ -28,14 +28,11 @@ ITensors.disable_warn_order()
   res_einexprs_exhaustive = contract(tn; sequence=seq_einexprs_exhaustive)[]
   seq_einexprs_greedy = contraction_sequence(tn; alg="einexpr", optimizer=Greedy())
   res_einexprs_greedy = contract(tn; sequence=seq_einexprs_exhaustive)[]
-  seq_einexprs_kahypar = contraction_sequence(tn; alg="einexpr", optimizer=HyPar())
-  res_einexprs_kahypar = contract(tn; sequence=seq_einexprs_kahypar)[]
   @test res_greedy ≈ res_optimal
   @test res_tree_sa ≈ res_optimal
   @test res_sa_bipartite ≈ res_optimal
   @test res_einexprs_exhaustive ≈ res_optimal
   @test res_einexprs_greedy ≈ res_optimal
-  @test res_einexprs_kahypar ≈ res_optimal
 
   if !Sys.iswindows()
     # KaHyPar doesn't work on Windows
@@ -46,5 +43,8 @@ ITensors.disable_warn_order()
     seq_kahypar_bipartite = contraction_sequence(tn; alg="kahypar_bipartite", sc_target=200)
     res_kahypar_bipartite = contract(tn; sequence=seq_kahypar_bipartite)[]
     @test res_optimal ≈ res_kahypar_bipartite
+    seq_einexprs_kahypar = contraction_sequence(tn; alg="einexpr", optimizer=HyPar())
+    res_einexprs_kahypar = contract(tn; sequence=seq_einexprs_kahypar)[]
+    @test res_einexprs_kahypar ≈ res_optimal
   end
 end