torch2trt icon indicating copy to clipboard operation
torch2trt copied to clipboard

[converters/stack.py] Update stack converter to support dynamic shapes. Resolves #830.

Open chaoz-dev opened this issue 2 years ago • 1 comments

Resolves issue #830, where torch.stack does not currently correctly handle dynamic shapes during TRT conversion.

Checked with included test:

python torch2trt/converters/stack_test.py

chaoz-dev avatar Jan 24 '23 19:01 chaoz-dev

Thanks for the PR!

I believe when I was working on dynamic shapes for either stack or chunk, I had an issue where adding dynamic shape support broke static models.

It's been a while so I forget what the concern was exactly, but I may want to sanity check a few things first.

jaybdub avatar Apr 28 '23 17:04 jaybdub