From f1e7d3110eecfb69cd29581f184332d857ba71bb Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sergio=20S=C3=A1nchez=20Ram=C3=ADrez?= Date: Thu, 28 Dec 2023 11:31:29 +0100 Subject: [PATCH] Fix KaHyPar tests --- test/KaHyPar_test.jl | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/test/KaHyPar_test.jl b/test/KaHyPar_test.jl index f4cdc84..0794b45 100644 --- a/test/KaHyPar_test.jl +++ b/test/KaHyPar_test.jl @@ -9,10 +9,11 @@ EinExpr([:i, :h, :d], Dict(i => 2 for i in [:i, :h, :d])), EinExpr([:d, :g, :c], Dict(i => 2 for i in [:d, :g, :c])), ] + sexpr = sum(tensors) - path = einexpr(HyPar(imbalance=0.42), EinExpr(Symbol[], tensors)) + path = einexpr(HyPar(imbalance = 0.42), sexpr) - @test path isa EinExpr + @test path isa SizedEinExpr @test mapreduce(flops, +, Branches(path)) == 108 end @@ -40,10 +41,11 @@ EinExpr([:A, :W], Dict(:A => 6, :W => 6)), EinExpr([:a, :C, :d], Dict(:a => 3, :d => 6, :C => 4)), ] + sexpr = sum(tensors) - path = einexpr(HyPar(imbalance=0.45), EinExpr(Symbol[], tensors)) + path = einexpr(HyPar(imbalance = 0.45), sexpr) - @test path isa EinExpr + @test path isa SizedEinExpr @test mapreduce(flops, +, Branches(path)) == 19099592 end