Skip to content

One-shot bufferization for onnx dialect #3355

@tungld

Description

@tungld

Do bufferization in MLIR style for onnx dialect, where we provide an one-short bufferization for onnx dialect to bufferize tensors, which has several advantages:

  • Once tensors are bufferized, the relationship between dynamic dimensions is more obvious via memref, which is benefical to the lowering of onnx ops
  • It is easy to make onnx-to-krnl work well with Linalg dialect, so that we can selectively low some onnx ops to Linalg when needed.

More information about bufferization in MLIR: https://mlir.llvm.org/docs/Bufferization/

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions