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.