Class: Kumi::IR::DF::Passes::BroadcastSimplify

Inherits:
Passes::Base
  • Object
show all
Defined in:
lib/kumi/ir/df/passes/broadcast_simplify.rb

Instance Method Summary collapse

Instance Method Details

#run(graph:, context: {}) ⇒ Object



10
11
12
13
# File 'lib/kumi/ir/df/passes/broadcast_simplify.rb', line 10

def run(graph:, context: {})
  functions = graph.functions.values.map { |fn| rewrite_function(fn) }
  Kumi::IR::DF::Graph.new(name: graph.name, functions: functions)
end