TensorRT
TensorRT copied to clipboard
NVIDIA
[Issue still exists in TRT 10.1] Fail to build the engine if input tensor number > 2^31
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.
