Anton Reinhard
b7560685d4
Reviewed-on: Rubydragon/MetagraphOptimization.jl#19 Co-authored-by: Anton Reinhard <anton.reinhard@proton.me> Co-committed-by: Anton Reinhard <anton.reinhard@proton.me>
31 lines
948 B
Julia
31 lines
948 B
Julia
"""
|
|
ReductionOptimizer
|
|
|
|
An optimizer that simply applies an available [`NodeReduction`](@ref) on each step. It implements [`optimize_to_fixpoint`](@ref). The fixpoint is reached when there are no more possible [`NodeReduction`](@ref)s in the graph.
|
|
"""
|
|
struct ReductionOptimizer <: AbstractOptimizer end
|
|
|
|
function optimize_step!(optimizer::ReductionOptimizer, graph::DAG)
|
|
# generate all options
|
|
operations = get_operations(graph)
|
|
if fixpoint_reached(optimizer, graph)
|
|
return false
|
|
end
|
|
|
|
push_operation!(graph, first(operations.nodeReductions))
|
|
|
|
return true
|
|
end
|
|
|
|
function fixpoint_reached(optimizer::ReductionOptimizer, graph::DAG)
|
|
operations = get_operations(graph)
|
|
return isempty(operations.nodeReductions)
|
|
end
|
|
|
|
function optimize_to_fixpoint!(optimizer::ReductionOptimizer, graph::DAG)
|
|
while !fixpoint_reached(optimizer, graph)
|
|
optimize_step!(optimizer, graph)
|
|
end
|
|
return nothing
|
|
end
|