Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Assertion while handling rank 8 tensors #441

Open
navahgar opened this issue Dec 11, 2024 · 0 comments
Open

Assertion while handling rank 8 tensors #441

navahgar opened this issue Dec 11, 2024 · 0 comments
Labels
mlir-tensorrt Pull request for the mlir-tensorrt project

Comments

@navahgar
Copy link

When tensors of rank 8 are present, it hits an assertion here.

assert(arrayRef.size() < nvinfer1::Dims::MAX_DIMS

Shouldn't we be able to allow tensors of rank == MAX_DIMS (which I believe is 8)?

@navahgar navahgar changed the title Handle Rank 8 tensors Assertion while handling rank 8 tensors Dec 11, 2024
@pranavm-nvidia pranavm-nvidia added the mlir-tensorrt Pull request for the mlir-tensorrt project label Dec 18, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
mlir-tensorrt Pull request for the mlir-tensorrt project
Projects
None yet
Development

No branches or pull requests

2 participants