castigli wrote: My PR will fix `mlir/test/Examples/NVGPU/Ch*.py` . I haven't looked at `mlir/test/Integration/GPU/CUDA/sm90/python/tools/matmulBuilder.py` but I assume that is still broken. https://github.com/llvm/llvm-project/pull/163883