nod-ai / SHARK-ModelDev

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

onnx.CumSum #662

Open jinchen62 opened 5 months ago

jinchen62 commented 5 months ago
jinchen62 commented 5 months ago

https://github.com/llvm/torch-mlir/pull/3241 adds OnnxToTorch support for attributes. The tests now have TorchToLinalg issue of torch.aten.cumsum op.

jinchen62 commented 5 months ago

Move to https://github.com/nod-ai/SHARK-Turbine/issues/347