Class: Kumi::IR::Vec::Passes::StencilDetection

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

Instance Method Summary collapse

Instance Method Details

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



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

def run(graph:, context: {})
  new_functions = graph.functions.values.map { tag_function(_1) }
  Kumi::IR::Vec::Module.new(name: graph.name, functions: new_functions)
end