Operator spec of Split operator's attribute split is wrong
@brb-nv is already working on this.
Since May 24, 2024.
- Dominant language
- C++
- Stars
- 13.4k
- Forks
- 2.4k
- Avg merge
- 5d 3h
- Merged PRs (30d)
- 2
Description
Description
My model has a split operator that is expected to divide each sub-tensor according to the dimensions specified.
As depicted in this setting:
This is the split operator link.
But each sub-tensor was divided equally using tensorrt, that is to say the split setting didn't take effect.
The error information is as follows:
[shapeContext.cpp::operator()::4017] Error Code 4: Shape Error (reshape wildcard -1 has no integer solution. Reshaping [1,18720,8,32] to [1,4990,-1].)
4990 is 24953 divided by 5.
Environment
TensorRT Version: 10.0
NVIDIA GPU: GeForce RTX 3090
NVIDIA Driver Version: 353
CUDA Version: 12.2
CUDNN Version:
Operating System: ubuntu22
Python Version (if applicable): 3.9.7
PyTorch Version (if applicable): 2.2
Baremetal or Container (if so, version): nvcr.io/nvidia/cuda:12.2.2-cudnn8-devel-ubuntu22.04
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.