Hi, I've just came across a model that requires support for ONNX CumSum op 
https://github.com/onnx/onnx/blob/master/docs/Operators.md#CumSum. The model 
comes from DETR object detection model 
https://github.com/facebookresearch/detr. Since this model doesn't need ad hoc 
object detection ops that are painful to support, I think it is a great fit for 
TVM. Our ONNX frontend (also PyTorch) only needs to implement Cumsum op.

Since TVM has support for scan operation 
https://tvm.apache.org/docs/tutorials/language/scan.html#sphx-glr-tutorials-language-scan-py,
 I'm wondering if it is a good idea to implement Relay cumsum op on top of te 
scan, or implement a new topi operator from scratch. I also want to utilize 
scan primitive from thrust to support fast cumsum on cuda.

@tqchen @kevinthesun @Laurawly @jwfromm





---
[Visit 
Topic](https://discuss.tvm.apache.org/t/supporting-cumsum-from-onnx-use-te-scan-op-or-develop-from-scratch/7830/1)
 to respond.

You are receiving this because you enabled mailing list mode.

To unsubscribe from these emails, [click 
here](https://discuss.tvm.apache.org/email/unsubscribe/b1f0ea64da5c455627578bc564be6dcea38fc140d3c5c8902e08e38aa6daa68c).

Reply via email to