GTR T5 Large Model output failure in TensorRT 8.6.1 when running on fp16 precision
@zerollzeng is already working on this.
Since Aug 12, 2023.
- Dominant language
- C++
- Stars
- 13.4k
- Forks
- 2.4k
- Avg merge
- 5d 3h
- Merged PRs (30d)
- 2
Description
Description
When I tried to convert gtr t5 model using trtexec command in the TensorRT docker container, the model is getting converted but is giving zero outputs, With fp32 there is no issue.
Environment
TensorRT Version: 8.6.1
NVIDIA GPU: A100
NVIDIA Driver Version: 525.85.12
CUDA Version: 12.1
CUDNN Version:
Operating System:
Python Version (if applicable): 3.10
Tensorflow Version (if applicable):
PyTorch Version (if applicable): 2.0.1
Baremetal or Container (if so, version): 23.07-py3 tensorrt
Steps To Reproduce
- Convert gtr-t5-large model from Hugging face to onnx using optimum-cli
- Convert the model from onnx to trt using trtexec with fp16 flag enabled.
- Infer the text, and we will get zeroes.
As most likely the issue is with overflow LayerNorm, Tried configurations like precisionConstraints (prefer, obey), converting to onnx using latest opset but still the output is coming as 0's and the issue persists.
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.