[Issue still exists in TRT 10.1] Fail to build the engine if input tensor number > 2^31

[dzzhang96]

Description

The release note of 10.0.1 mentioned that the issue “UNets with tensors containing >2^31 elements may fail during the engine building step” was fixed. However, my model (nnUNet) has 1x32x512x512x512 input and it still failed to build the engine in TRT 10.1.

Environment

TensorRT Version: 10.1

Relevant Files

Model link: I uploaded a dummy model for testing here.

[lix19937]

Similar case #3815

[dzzhang96]

any updates? :)

[moraxu]

Closing in favor of #4123