Skip to content

Commit

Permalink
Fix KaHyPar test
Browse files Browse the repository at this point in the history
  • Loading branch information
jofrevalles committed Nov 3, 2023
1 parent c92f8bf commit 628671a
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions test/KaHyPar_test.jl
Original file line number Diff line number Diff line change
Expand Up @@ -41,10 +41,10 @@
EinExpr([:a, :C, :d], Dict(:a => 3, :d => 6, :C => 4)),
]

path = einexpr(HyPar, EinExpr(Symbol[], tensors))
path = einexpr(HyPar(imbalance=0.45), EinExpr(Symbol[], tensors))

@test path isa EinExpr

@test mapreduce(flops, +, Branches(path)) == 31653164
@test mapreduce(flops, +, Branches(path)) == 19099592
end
end

0 comments on commit 628671a

Please sign in to comment.