mlir/test/Transforms/test-legalize-target-materialization-no-uses.mlir

   1 // RUN: mlir-opt -test-target-materialization-with-no-uses -split-input-file %s | FileCheck %s
   2
   3 // The conversion is set up as follows:
   4 // - type_changer ops are illegal;
   5 // - type_changer ops are replaced with their operands;
   6 // - i16 types are converted to i64 by the type conversion;
   7 // - the rest of the types are legal.
   8 // The first type_changer is replaced with its operand. For the pattern to
   9 // apply to the second type_changer, the conversion infra creates a dummy
  10 // cast operation to cast from the i32 to i64 because the original op takes an
  11 // (illegal) i16 that became i64. This dummy operation should be replaced by
  12 // the one produced by the target materialization hook. At the moment when the
  13 // materialization decision is taken, the i64 replacement of the first type
  14 // change (the result of the dummy cast) has no uses, but the value it replaces
  15 // does, so the infra must call the materialization rather than assume the
  16 // dummy cast to be dead.
  17
  18 // CHECK-LABEL: @foo
  19 func.func @foo() {
  20   %0 = "test.type_producer"() : () -> i32
  21   // CHECK: test.cast
  22   // CHECK-NOT: test.type_changer
  23   %1 = "test.type_changer"(%0) : (i32) -> i16
  24   %2 = "test.type_changer"(%1) : (i16) -> i64
  25   "test.type_consumer"(%2) : (i64) -> ()
  26   return
  27 }
  28
  29 // -----
  30
  31 // CHECK-LABEL: @direct_forward
  32 func.func @direct_forward(%arg0 : i16) -> i16 {
  33   // CHECK-NEXT: return
  34   %0 = "test.type_changer"(%arg0) : (i16) -> i16
  35   return %0 : i16
  36 }