mlir/test/Transforms/test-rewrite-dynamic-op.mlir


1
2
3
4
5
6
7
8
9
10
11
12

// RUN: mlir-opt %s -test-rewrite-dynamic-op | FileCheck %s

// Test that `test.one_operand_two_results` is replaced with
// `test.generic_dynamic_op`.

// CHECK-LABEL: func @rewrite_dynamic_op
func.func @rewrite_dynamic_op(%arg0: i32) {
  // CHECK-NEXT: %{{.*}}:2 = "test.dynamic_generic"(%arg0) : (i32) -> (i32, i32)
  %0:2 = "test.dynamic_one_operand_two_results"(%arg0) : (i32) -> (i32, i32)
  // CHECK-NEXT: return
  return
}