onnx-mlir icon indicating copy to clipboard operation
onnx-mlir copied to clipboard

Published 20 hours ago • verified publisher icon onnx

Loop collapsing with fusion

Open chentong319 opened this issue 1 year ago • 0 comments

This PR handles loop collapsing in op fusion. There are two parts:

  1. Decide which loops to be collapsed. When ops are fused, all its operands should be checked whether its dimensions are involved in broadcasting. Such information will be used to decide which loops should be collapsed.
  2. Code generation when some loops are collapsed.

chentong319 avatar Nov 13 '23 22:11 chentong319