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

[TorchToLinalg] Adds Quantization Support for ConvTranspose #3240

Merged
merged 8 commits into from
Apr 30, 2024

Conversation

zjgarvey
Copy link
Collaborator

I spent a little while debugging numerics issues with some tests similar to the ones in quantized_models.py, only to find that pytorch's quantized conv transpose is catastrophically inaccurate. I'll upstream the issue and only leave the tests here which are of the form quantize -> dequantize -> op.

@zjgarvey zjgarvey changed the title Adds Quantization Support for ConvTranspose [TorchToLinalg] Adds Quantization Support for ConvTranspose Apr 26, 2024
@zjgarvey zjgarvey requested a review from rsuderman April 29, 2024 14:44
lib/Conversion/TorchToLinalg/Linear.cpp Outdated Show resolved Hide resolved
@rsuderman rsuderman merged commit 72349f7 into llvm:main Apr 30, 2024
3 checks passed
archana-ramalingam pushed a commit to archana-ramalingam/torch-mlir that referenced this pull request May 8, 2024
I spent a little while debugging numerics issues with some tests similar
to the ones in quantized_models.py, only to find that pytorch's
quantized conv transpose is catastrophically inaccurate. I'll upstream
the issue and only leave the tests here which are of the form quantize
-> dequantize -> op.
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.

2 participants