[mlir][py] Add StrAttr convenience builder.

diff --git a/mlir/python/mlir/ir.py b/mlir/python/mlir/ir.py
index 82468e8..1e24fcb 100644 (file)
--- a/mlir/python/mlir/ir.py
+++ b/mlir/python/mlir/ir.py
@@ -32,6 +32,10 @@ def _i64Attr(x, context):
   return IntegerAttr.get(
       IntegerType.get_signless(64, context=context), x)
 
+@register_attribute_builder("StrAttr")
+def _stringAttr(x, context):
+  return StringAttr.get(x, context=context)
+
 @register_attribute_builder("SymbolNameAttr")
 def _symbolNameAttr(x, context):
   return StringAttr.get(x, context=context)

diff --git a/mlir/test/python/dialects/shape.py b/mlir/test/python/dialects/shape.py
index 2d2a203..3e7a8b2 100644 (file)
--- a/mlir/test/python/dialects/shape.py
+++ b/mlir/test/python/dialects/shape.py
@@ -25,11 +25,14 @@ def testConstShape():
         shape.ConstWitnessOp(False)
         shape.ConstSizeOp(30)
         shape.ConstSizeOp(IntegerAttr.get(IndexType.get(), 40))
-        shape.ConstShapeOp([1, 2])
+        x = shape.ConstShapeOp([1, 2])
+        shape.MeetOp(x, x, error="impossible")
         return shape.ConstShapeOp(
             DenseElementsAttr.get(
                 np.array([3, 4], dtype=np.int64), type=IndexType.get()))
 
+
+
     # CHECK-LABEL: func @const_shape_tensor(%arg0: tensor<12x?xf32>)
     # CHECK-DAG: shape.const_witness false
     # CHECK-DAG: shape.const_size 30