durga4github wrote: The wmma.py change in the NVPTX backend looks good to me. But I would still prefer it as a separate PR with an [NVPTX] tag. @Artem-B , Please let us know what do you think. https://github.com/llvm/llvm-project/pull/94035