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

Refine PT2ONNX export #901

Merged
merged 17 commits into from
May 29, 2023
Merged

Refine PT2ONNX export #901

merged 17 commits into from
May 29, 2023

Conversation

yuwenzho
Copy link
Contributor

@yuwenzho yuwenzho commented May 23, 2023

Type of Change

feature
API not changed

Description

Refine PT2ONNX export

old PT2ONNX export workflow:

  • step 1: get quantized ops + scale & zp info from INT8 PT
  • step 2: export FP32 PT to FP32 ONNX, then quantize to INT8 ONNX with quantized ops info collected in step1
  • step 3: map scale & zp info collected in step1 to INT8 ONNX

new PT2ONNX export workflow:

  • INT8 PT --> model wrapper (to unify model output) --> export to INT8 ONNX

Expected Behavior & Potential Risk

pass extension test of export

How has this PR been tested?

extension test of export

Dependency Change?

no

yuwenzho added 8 commits May 17, 2023 14:30
Signed-off-by: yuwenzho <[email protected]>
Signed-off-by: yuwenzho <[email protected]>
Signed-off-by: yuwenzho <[email protected]>
Signed-off-by: yuwenzho <[email protected]>
Signed-off-by: yuwenzho <[email protected]>
Signed-off-by: yuwenzho <[email protected]>
@yuwenzho yuwenzho requested a review from xin3he May 23, 2023 01:26
@xin3he xin3he requested review from hshen14, ftian1 and PenghuiCheng May 23, 2023 01:37
@chensuyue chensuyue added enhancement New feature or request extension test labels May 23, 2023
@chensuyue chensuyue added this to the v2.2 milestone May 23, 2023
yuwenzho added 4 commits May 23, 2023 14:06
Signed-off-by: yuwenzho <[email protected]>
Signed-off-by: yuwenzho <[email protected]>
Signed-off-by: yuwenzho <[email protected]>
Signed-off-by: yuwenzho <[email protected]>
Copy link
Contributor

@hshen14 hshen14 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@yuwenzho
Copy link
Contributor Author

@yuwenzho
Copy link
Contributor Author

/Azurepipeline run UT-Basic

@azure-pipelines
Copy link

Azure Pipelines successfully started running 1 pipeline(s).

@yuwenzho
Copy link
Contributor Author

/Azurepipeline run

@azure-pipelines
Copy link

Azure Pipelines successfully started running 5 pipeline(s).

@chensuyue chensuyue merged commit 877adbd into master May 29, 2023
@chensuyue chensuyue deleted the yuwenzho/export branch May 29, 2023 01:10
yiliu30 pushed a commit to yiliu30/neural-compressor that referenced this pull request Feb 28, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request extension test
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants