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

ONNXToDeepSparse matmul integer conversion patches for whipser support #1616

Merged
merged 4 commits into from
Jun 13, 2023

Conversation

bfineran
Copy link
Member

@bfineran bfineran commented Jun 8, 2023

Quantization experiments for the whipser model resulted with a few MatMulInteger ops that did not have their weights properly folded. This is primarily due to the non-parameterized matmul matching improperly not skipping over parameterized matmuls (see fixes in src/sparseml/exporters/transforms/matmul_to_matmulinteger_cast_mul.py).

The longer fixes in the other two files update pattern matching to convert parameterized matmuls without bias adds, this were lifted from my updates in #1559 that are still pending review. We can merge these first as they are tangential to that PR as well

test_plan:
Improper conversion before:
Screenshot 2023-06-08 at 5 20 05 PM

Fixed conversion after:
Screenshot 2023-06-08 at 5 20 33 PM

@bfineran bfineran requested review from mgoin and KSGulin June 8, 2023 21:20
@bfineran bfineran self-assigned this Jun 8, 2023
dbogunowicz
dbogunowicz previously approved these changes Jun 11, 2023
KSGulin
KSGulin previously approved these changes Jun 12, 2023
Copy link
Contributor

@KSGulin KSGulin left a comment

Choose a reason for hiding this comment

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

Looks like the exporter tests are failing. Otherwise LGTM!

@bfineran bfineran dismissed stale reviews from KSGulin and dbogunowicz via 92b25ff June 13, 2023 15:50
@bfineran bfineran merged commit ce0aaea into main Jun 13, 2023
10 checks passed
@bfineran bfineran deleted the whisper-onnx-to-deepsparse-patches branch June 13, 2023 16:02
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

Successfully merging this pull request may close these issues.

None yet

3 participants