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

how to convert int8 converted onnx model to tensorrt? #20

Closed
lucasjinreal opened this issue Mar 16, 2022 · 1 comment
Closed

how to convert int8 converted onnx model to tensorrt? #20

lucasjinreal opened this issue Mar 16, 2022 · 1 comment

Comments

@lucasjinreal
Copy link

Hi, I followed the example and successfully converted the model to int8 on GPT2 model.
but the model generated with some Customized onnx op:

Shape,Gather,Range,Unsqueeze,Concat,Reshape,Add,LayerNormalization,DynamicQua │
│ ntizeLinear,Slice,Mul,MatMulInteger,Squeeze,Cast,Split,Sub,Transpose,MatMul,Pow,Div,Wh │
│ ere,Softmax,FastGelu,SkipLayerNormalization

Such as DynamicQuantizeLinear, FastGelu, how to converted it to tensorrt?

the int8 model was 400M compares with original 1.2G. much more small, if can inference via tensorrt, it can be massively accelerated.

@ykim362
Copy link
Member

ykim362 commented May 18, 2022

In this repository, we demonstrate CPU quantization only. For the GPU quantization, please visit https://github.com/Microsoft/onnxruntime. Thank you for your interest.

@ykim362 ykim362 closed this as completed May 18, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants