rubydragon/metagraphoptimization.jl
1
0
Fork 0
Code Issues 6 Pull Requests Actions Packages Projects Releases Wiki Activity

experiments (#1)
All checks were successful
MetagraphOptimization_CI / docs (push) Successful in 10m41s
Details
MetagraphOptimization_CI / test (push) Successful in 30m40s
Details

Browse Source 
Co-authored-by: Anton Reinhard <anton.reinhard@proton.me>
Reviewed-on: #1
This commit is contained in:
rubydragon
2024-05-08 12:03:27 +02:00
parent 82ed774b7e
commit 87dbaf2c32
155 changed files with 5372 additions and 1029 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
test/unit_tests_optimization.jl
View File

@ -6,7 +6,8 @@ RNG = Random.default_rng()
graph = parse_dag(joinpath(@__DIR__, "..", "input", "AB->ABBB.txt"), ABCModel())
# create the optimizers
FIXPOINT_OPTIMIZERS = [GreedyOptimizer(GlobalMetricEstimator()), ReductionOptimizer()]
FIXPOINT_OPTIMIZERS =
[GreedyOptimizer(GlobalMetricEstimator()), ReductionOptimizer(), SplitOptimizer(), FusionOptimizer()]
NO_FIXPOINT_OPTIMIZERS = [RandomWalkOptimizer(RNG)]
@testset "Optimizer $optimizer" for optimizer in vcat(NO_FIXPOINT_OPTIMIZERS, FIXPOINT_OPTIMIZERS)
@ -16,7 +17,7 @@ NO_FIXPOINT_OPTIMIZERS = [RandomWalkOptimizer(RNG)]
@test !fixpoint_reached(optimizer, graph)
@test operation_stack_length(graph) == 1
@test optimize!(optimizer, graph, 10)
@test optimize!(optimizer, graph, 2)
@test !fixpoint_reached(optimizer, graph)