Uh oh!
There was an error while loading. Please reload this page.
Add CPUID for AvxVnniInt8 and AvxVnniInt16 - #113956
Conversation
6714249 to
141d643Comparekhushal1996
commented
Mar 27, 2025
@tannergooding This is first of the 2 PRs needed for AVX VNNI INT* API introduction #112586 |
Uh oh!
There was an error while loading. Please reload this page.
Uh oh!
There was an error while loading. Please reload this page.
141d643 to
98fc970Comparekhushal1996
commented
Apr 14, 2025
@tannergooding@saucecontrol I have added the CPUID, API surface, JIT handling and template tests here. |
Uh oh!
There was an error while loading. Please reload this page.
e6cf454 to
90fa072CompareUh oh!
There was an error while loading. Please reload this page.
Uh oh!
There was an error while loading. Please reload this page.
Uh oh!
There was an error while loading. Please reload this page.
tannergooding
left a comment
There was a problem hiding this comment.
LGTM, minus a couple nits on ways to simplify the diffs.
CC. @dotnet/jit-contrib for secondary review
Co-authored-by: Tanner Gooding <tagoo@outlook.com>
EgorBo
commented
Jul 1, 2025
@khushal1996 could you please resolve the merge conflict? |
Uh oh!
There was an error while loading. Please reload this page.
tannergooding
commented
Jul 6, 2025
Resolved the merge conflicts. I'll get this merged once CI completes |
khushal1996
commented
Jul 6, 2025
Thanks @tannergooding for picking this up. Let me know if you want me to take over if the CI fails. |
Uh oh!
There was an error while loading. Please reload this page.
khushal1996
commented
Jul 7, 2025
Thanks for helping with the review and fixes @tannergooding |
tannergooding
commented
Jul 7, 2025
Thanks as well for getting the feature implemented and done @khushal1996. Fixes were just some minor merge conflict fixes from other cleanup/simplifications that had happened |
This PR adds support for CPUID for
AVX-VNNI-INT8&AVX-VNNI-INT16ISAsThis resolves#112586
Design
The changes are made in a way to enable the 2 ISAs when
Avx10.2is enabled orThis is w.r.t the discussions done in API proposal #112586
Testing
Note1: Emitter unit tests not ran since they are added and verified along with AVX10.2 PR #111209
Note2: Superpmi results are not accurate since we are adding a new CPUID and it leads to a new jiteeversionguid. Even after changing the jiteeversion manually, superpmi run shows errors and failures based on the old mch files which can be ignored.
Run JIT subtree with AVXVNNIINT* enabled / disabled
AVXVNNIINT* Enabled

AVXVNNIINT* disabled
