onnx-tensorrt icon indicating copy to clipboard operation
onnx-tensorrt copied to clipboard

Is there any plan to support SequenceConstruct?

Open DoGunKIM93 opened this issue 2 years ago • 4 comments

Description

onnx-tensorrt/docs/operators.md still says don't support SequenceConstruct, SequenceAt, etc. So is any plan to support these Sequence ops?

Environment

TensorRT Version: 8.0.4 ONNX-TensorRT Version / Branch: main GPU Type: V100 Nvidia Driver Version: 470.103.01 CUDA Version: 11.6 CUDNN Version: 8.2.4 Operating System + Version: ubuntu 20.04 Python Version (if applicable): 3.8.12 TensorFlow + TF2ONNX Version (if applicable): PyTorch Version (if applicable): 1.11.0 Baremetal or Container (if container which image + tag): nvcr.io/nvidia/pytorch:22.04-py3

Relevant Files

Steps To Reproduce

DoGunKIM93 avatar Aug 04 '22 08:08 DoGunKIM93

There is no immediate plan to support sequence ops in TensorRT .

kevinch-nv avatar Aug 22 '22 16:08 kevinch-nv

I need one of three option to implement seq2seq decode model (use transformer):

  1. sequence ops
  2. IRecurrenceLayer support change size input when loop (time axis increase size when loop)
  3. Slice by iterator of loop (allocate max length and slice to use) @kevinch-nv Is there any plan to support seq2seq decode model?

anhtu812 avatar Nov 06 '23 07:11 anhtu812

@kevinch-nv @rajeevsrao : Basically, most of the loops would require a sequence.

borisfom avatar Nov 11 '23 01:11 borisfom

If anyone gets to this issue while trying to convert an ONNX model to TensorRT: TorchDynamo as of torch 2.2.2 converts torch.cat to this op, and if you switch to TorchScript instead you won't have this issue for the conversion.

ryanli avatar Apr 24 '24 17:04 ryanli