Quantized model test data on GPU
Nobody has claimed this yet.
- Dominant language
- Jupyter Notebook
- Stars
- 9.8k
- Forks
- 1.6k
- PR merge metrics
- No merged PRs in 30d
Description
Ask a Question
Since the GPU machines of CI have been upgraded from NV6 to T4, it looks quantized model on GPU should be added too.
Hardware support is required to achieve better performance with quantization on GPUs. You need a device that supports Tensor Core int8 computation, like T4 or A100.
https://onnxruntime.ai/docs/performance/quantization.html#quantization-on-gpu
But it looks that the test result on CPU with VNNI if different from on GPU. Is it expected? @yufenglee
If it's expected, shall we add test data on GPU? @jcwchen @snnn
Contributor guide
No contributing guide indexed for this repository
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.
Research direction
Review the ONNX Runtime quantization-on-GPU documentation and the linked Azure Pipelines test result first, then compare the CPU/VNNI and T4 outcomes. Done means establishing whether the difference is expected and deciding whether GPU-specific quantized-model test data should be added.
Written by the indexing model from the issue text.
Assessment
- Domain
- machine-learning, testing
- Issue type
- Feature
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 30/100