37 lines
1.1 KiB
Julia
37 lines
1.1 KiB
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.
|
|
|
|
See also: [`FusionOptimizer`](@ref), [`SplitOptimizer`](@ref)
|
|
"""
|
|
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
|
|
|
|
function String(::ReductionOptimizer)
|
|
return "reduction_optimizer"
|
|
end
|