[mlir] Improve test coverage for print-op-graph

llvm · Feb 27, 2021 · dd2f50a · dd2f50a
1 parent 6a337f8
commit dd2f50a
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/mlir/test/Transforms/print-op-graph.mlir b/mlir/test/Transforms/print-op-graph.mlir
@@ -1,12 +1,19 @@
-// RUN: mlir-opt -allow-unregistered-dialect -print-op-graph %s -o %t 2>&1 | FileCheck %s
+// RUN: mlir-opt -allow-unregistered-dialect -mlir-elide-elementsattrs-if-larger=2 -print-op-graph %s -o %t 2>&1 | FileCheck %s
 
 // CHECK-LABEL: digraph "merge_blocks"
+// CHECK{LITERAL}: value: [[...]] : tensor\<2x2xi32\>}
+// CHECK{LITERAL}: value: dense\<1\> : tensor\<5xi32\>}
+// CHECK{LITERAL}: value: dense\<[[0, 1]]\> : tensor\<1x2xi32\>}
 func @merge_blocks(%arg0: i32, %arg1 : i32) -> () {
-  %0:2 = "test.merge_blocks"() ({
+  %0 = constant dense<[[0, 1], [2, 3]]> : tensor<2x2xi32>
+  %1 = constant dense<1> : tensor<5xi32>
+  %2 = constant dense<[[0, 1]]> : tensor<1x2xi32>
+
+  %3:2 = "test.merge_blocks"() ({
   ^bb0:
      "test.br"(%arg0, %arg1)[^bb1] : (i32, i32) -> ()
   ^bb1(%arg3 : i32, %arg4 : i32):
      "test.return"(%arg3, %arg4) : (i32, i32) -> ()
   }) : () -> (i32, i32)
-  "test.return"(%0#0, %0#1) : (i32, i32) -> ()
+  "test.return"(%3#0, %3#1) : (i32, i32) -> ()
 }