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

[MLIR][TORCH] Add OnnxToLinalg lowering pipeline

Open vivekkhandelwal1 opened this issue 1 year ago • 1 comments

This commit adds the lowering pipeline for the onnx-torch IR to linalg-on-tensors backend IR. The pipeline comprises of three stages: 1.) OnnxToTorch Lowering Pass 2.) TorchFunctionToTorchBackend Pipeline 3.) TorchBackendToLinalgOnTensorsBackend Pipeline

Signed-Off By: Vivek Khandelwal [email protected]

vivekkhandelwal1 avatar Jan 31 '24 15:01 vivekkhandelwal1

What is this doing above and beyond the normal torch to linalg pipeline?

I'd rather just teach that one what to do vs having another bespoke pipeline.

(Or if just adding a pass at the front door onnx, delegate to the regular one)

stellaraccident avatar Feb 01 '24 16:02 stellaraccident

This PR is not required anymore therefore closing it.

vivekkhandelwal1 avatar Apr 02 '24 10:04 vivekkhandelwal1