Improve tile_buf asm printing and add tests

Author: PTOAS

lib/PTO/IR/PTOTypeDefs.cpp

@@ -0,0 +1,26 @@
+// RUN: ptoas --print-ir-after-all --print-ir-after-all-func-filter=tilebuf_asm --pto-level=level3 %s 2>&1 1>/dev/null | FileCheck %s
+
+module {
+  func.func @tilebuf_asm(
+      %arg0: !pto.tile_buf<loc=vec, dtype=f32, rows=1, cols=16, v_row=1, v_col=16,
+                           blayout=row_major, slayout=none_box, fractal=512, pad=0>,
+      %arg1: !pto.tile_buf<vec, 16x128xf32, valid=16x1, blayout=col_major,
+                           slayout=row_major, fractal=1024, pad=2>) {
+    return
+  }
+
+  func.func @tilebuf_asm_dynamic(
+      %arg0: !pto.tile_buf<loc=vec, dtype=f16, rows=32, cols=32, v_row=?, v_col=?,
+                           blayout=row_major, slayout=none_box, fractal=512, pad=0>,
+      %arg1: !pto.tile_buf<acc, 16x16xbf16, valid=?x8, blayout=col_major>) {
+    return
+  }
+}
+
+// CHECK: func.func @tilebuf_asm
+// CHECK: !pto.tile_buf<vec, 1x16xf32>
+// CHECK: !pto.tile_buf<vec, 16x128xf32, valid=16x1, blayout=col_major, slayout=row_major, fractal=1024, pad=2>
+
+// CHECK: func.func @tilebuf_asm_dynamic
+// CHECK: !pto.tile_buf<vec, 32x32xf16, valid=?x?>
+// CHECK: !pto.tile_buf<acc, 16x16xbf16, valid=?x8, blayout=col_major>

test/basic/tilebuf_asm_compact.mlir

@@ -0,0 +1,10 @@
+// RUN: ptoas --pto-level=level3 %s 2>&1 1>/dev/null | FileCheck %s
+
+module {
+  func.func @tilebuf_asm_cols_negative(
+      %arg0: !pto.tile_buf<loc=vec, dtype=f32, rows=1, cols=-1>) {
+    return
+  }
+}
+
+// CHECK: error: cols must be non-negative

test/basic/tilebuf_asm_error_cols_negative.mlir

@@ -0,0 +1,10 @@
+// RUN: ptoas --pto-level=level3 %s 2>&1 1>/dev/null | FileCheck %s
+
+module {
+  func.func @tilebuf_asm_conflicting_cols(
+      %arg0: !pto.tile_buf<vec, 16x16xf32, cols=8, rows=16>) {
+    return
+  }
+}
+
+// CHECK: error: conflicting cols values

test/basic/tilebuf_asm_error_conflicting_cols.mlir

@@ -0,0 +1,10 @@
+// RUN: ptoas --pto-level=level3 %s 2>&1 1>/dev/null | FileCheck %s
+
+module {
+  func.func @tilebuf_asm_conflicting_dtype(
+      %arg0: !pto.tile_buf<vec, 16x16xf32, dtype=f16>) {
+    return
+  }
+}
+
+// CHECK: error: conflicting dtype values

test/basic/tilebuf_asm_error_conflicting_dtype.mlir

@@ -0,0 +1,10 @@
+// RUN: ptoas --pto-level=level3 %s 2>&1 1>/dev/null | FileCheck %s
+
+module {
+  func.func @tilebuf_asm_conflicting_loc(
+      %arg0: !pto.tile_buf<vec, loc=acc, 16x16xf32>) {
+    return
+  }
+}
+
+// CHECK: error: conflicting loc values

test/basic/tilebuf_asm_error_conflicting_loc.mlir

@@ -0,0 +1,10 @@
+// RUN: ptoas --pto-level=level3 %s 2>&1 1>/dev/null | FileCheck %s
+
+module {
+  func.func @tilebuf_asm_conflicting_positional_dtype(
+      %arg0: !pto.tile_buf<vec, 16x16xf32, 16x16xbf16>) {
+    return
+  }
+}
+
+// CHECK: error: conflicting dtype values

test/basic/tilebuf_asm_error_conflicting_positional_dtype.mlir

@@ -0,0 +1,10 @@
+// RUN: ptoas --pto-level=level3 %s 2>&1 1>/dev/null | FileCheck %s
+
+module {
+  func.func @tilebuf_asm_conflicting_positional_shape(
+      %arg0: !pto.tile_buf<vec, 16x16xf32, 8x16xf32>) {
+    return
+  }
+}
+
+// CHECK: error: conflicting rows/cols values

test/basic/tilebuf_asm_error_conflicting_positional_shape.mlir

@@ -0,0 +1,10 @@
+// RUN: ptoas --pto-level=level3 %s 2>&1 1>/dev/null | FileCheck %s
+
+module {
+  func.func @tilebuf_asm_conflicting_rows(
+      %arg0: !pto.tile_buf<vec, 16x16xf32, rows=8, cols=16>) {
+    return
+  }
+}
+
+// CHECK: error: conflicting rows values

test/basic/tilebuf_asm_error_conflicting_rows.mlir

@@ -0,0 +1,10 @@
+// RUN: ptoas --pto-level=level3 %s 2>&1 1>/dev/null | FileCheck %s
+
+module {
+  func.func @tilebuf_asm_invalid_blayout(
+      %arg0: !pto.tile_buf<vec, 16x16xf32, blayout=bad>) {
+    return
+  }
+}
+
+// CHECK: error: unknown blayout: bad