search

nod-ai / SHARK-ModelDev

Unified compiler/runtime for interfacing with PyTorch Dynamo.
Apache License 2.0
95 stars 48 forks source link

Fix Onnx.DFT Torch->Linalg lowering #800

Open vivekkhandelwal1 opened 4 months ago

vivekkhandelwal1 commented 4 months ago

The Onnx.DFT lowers from Onnx->Torch but fails during Torch->Linalg lowering. The reason is that the Torch->Linalg lowering for AtenFftFftOp, AtenFftIfftOp is missing. In order to fix this issue, these lowerings have to be added.

kumardeepakamd commented 2 months ago

@PhaneeshB will need you to help out on few ONNX OPs like this one to meet our Sept goal

PhaneeshB commented 2 months ago

Will wait for this op to show up in models so we know the paths for which we want to add lowering. As of now, there aren't any models failing due to this op in #812