mlir/test/IR/print-op-custom-or-generic.mlir


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28

// # RUN: mlir-opt %s -split-input-file | FileCheck %s
// # RUN: mlir-opt %s -mlir-print-op-generic -split-input-file  | FileCheck %s --check-prefix=GENERIC

// Check that `printCustomOrGenericOp` and `printGenericOp` print the right
// assembly format. For operations without custom format, both should print the
// generic format.

// CHECK-LABEL: func @op_with_custom_printer
// CHECK-GENERIC-LABEL: "func"()
func.func @op_with_custom_printer() {
  %x = test.string_attr_pretty_name
  // CHECK: %x = test.string_attr_pretty_name
  // GENERIC: %0 = "test.string_attr_pretty_name"()
  return
  // CHECK: return
  // GENERIC: "func.return"()
}

// -----

// CHECK-LABEL: func @op_without_custom_printer
// CHECK-GENERIC: "func"()
func.func @op_without_custom_printer() {
  // CHECK: "test.result_type_with_trait"() : () -> !test.test_type_with_trait
  // GENERIC: "test.result_type_with_trait"() : () -> !test.test_type_with_trait
  "test.result_type_with_trait"() : () -> !test.test_type_with_trait
  return
}