[Mlir-commits] [mlir] [mlir][Python] fix examples after #172892 (PR #174481)

[llvmlistbot at llvm.org]

llvmbot wrote:


<!--LLVM PR SUMMARY COMMENT-->

@llvm/pr-subscribers-mlir-gpu

Author: Maksim Levental (makslevental)

<details>
<summary>Changes</summary>



---
Full diff: https://github.com/llvm/llvm-project/pull/174481.diff


2 Files Affected:

- (modified) mlir/test/Examples/NVGPU/tools/nvdsl.py (+4-4) 
- (modified) mlir/test/Integration/GPU/CUDA/sm90/python/tools/matmulBuilder.py (+5-5) 


``````````diff
diff --git a/mlir/test/Examples/NVGPU/tools/nvdsl.py b/mlir/test/Examples/NVGPU/tools/nvdsl.py
index 856107293470d..53e1db011c7de 100644
--- a/mlir/test/Examples/NVGPU/tools/nvdsl.py
+++ b/mlir/test/Examples/NVGPU/tools/nvdsl.py
@@ -15,21 +15,21 @@
 def const(value: int, ty=None):
     ty = T.index() if ty is None else ty
     if isinstance(value, ir.Value) and (
-        value.type.isinstance(value.type) or T.bool().isinstance(value.type)
+        isinstance(value.type, value.type) or isinstance(value.type, ir.BoolType)
     ):
         return value
     return arith.constant(ty, value)
 
 
 def get_type_size(ty):
-    if ir.MemRefType.isinstance(ty):
+    if isinstance(ir.MemRefType, ty):
         size = get_type_size(ty.element_type)
         for sz in ty.shape:
             size *= sz
         return size
-    if ir.FloatType.isinstance(ty):
+    if isinstance(ir.FloatType, ty):
         return ir.FloatType(ty).width // 8
-    if ir.IntegerType.isinstance(ty):
+    if isinstance(ir.IntegerType, ty):
         return ir.IntegerType(ty).width // 8
     raise NotImplementedError(ty)
 
diff --git a/mlir/test/Integration/GPU/CUDA/sm90/python/tools/matmulBuilder.py b/mlir/test/Integration/GPU/CUDA/sm90/python/tools/matmulBuilder.py
index 5394d4a327255..e53fa885acc38 100644
--- a/mlir/test/Integration/GPU/CUDA/sm90/python/tools/matmulBuilder.py
+++ b/mlir/test/Integration/GPU/CUDA/sm90/python/tools/matmulBuilder.py
@@ -76,9 +76,9 @@ def debug_print(fmt, *args, predicate=None, threadNumber=-1, forcePrint=False):
     type_formats = []
     for arg in args:
         ty_format = None
-        if ir.IndexType.isinstance(arg.type):
+        if isinstance(ir.IndexType, arg.type):
             ty_format = "%llu"
-        if ir.IntegerType.isinstance(arg.type):
+        if isinstance(ir.IntegerType, arg.type):
             width = ir.IntegerType(arg.type).width
             if width == 64:
                 ty_format = "%llu"
@@ -86,7 +86,7 @@ def debug_print(fmt, *args, predicate=None, threadNumber=-1, forcePrint=False):
                 ty_format = "%d"
             elif width == 1:
                 ty_format = "%i"
-        if ir.F32Type.isinstance(arg.type):
+        if isinstance(ir.F32Type, arg.type):
             ty_format = "%f"
         if ty_format is None:
             raise NotImplementedError(arg.type)
@@ -102,9 +102,9 @@ def debug_print(fmt, *args, predicate=None, threadNumber=-1, forcePrint=False):
 
 
 def get_type_size(ty):
-    if ir.FloatType.isinstance(ty):
+    if isinstance(ir.FloatType, ty):
         return ir.FloatType(ty).width // 8
-    if ir.IntegerType.isinstance(ty):
+    if isinstance(ir.IntegerType, ty):
         return ir.IntegerType(ty).width // 8
     raise NotImplementedError(ty)
 

``````````

</details>


https://github.com/llvm/llvm-project/pull/174481