NVIDIA / NVIDIA/TensorRT

Operator spec of Split operator's attribute split is wrong

Open
#3,892 6 comments 0 reactions 1 assignee View on GitHub

@brb-nv is already working on this.

Since May 24, 2024.

triaged
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:

image

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

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.