diff options
| author | Diego Caballero <diego.caballero@intel.com> | 2019-11-19 10:15:36 -0800 |
|---|---|---|
| committer | A. Unique TensorFlower <gardener@tensorflow.org> | 2019-11-19 11:09:39 -0800 |
| commit | dd5a7cb48833d4abf93a063f40b7cf5baae940ef (patch) | |
| tree | 7e30e57c57499894dd1f75b12aa28fdaf298915f /mlir/test/Transforms | |
| parent | 06fb797b4090526b906d7af44715a826faed5d3a (diff) | |
| download | bcm5719-llvm-dd5a7cb48833d4abf93a063f40b7cf5baae940ef.tar.gz bcm5719-llvm-dd5a7cb48833d4abf93a063f40b7cf5baae940ef.zip | |
Add getRemappedValue to ConversionPatternRewriter
This method is needed for N->1 conversion patterns to retrieve remapped
Values used in the original N operations.
Closes tensorflow/mlir#237
COPYBARA_INTEGRATE_REVIEW=https://github.com/tensorflow/mlir/pull/237 from dcaballe:dcaballe/getRemappedValue 1f64fadcf2b203f7b336ff0c5838b116ae3625db
PiperOrigin-RevId: 281321881
Diffstat (limited to 'mlir/test/Transforms')
| -rw-r--r-- | mlir/test/Transforms/test-legalize-remapped-value.mlir | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/mlir/test/Transforms/test-legalize-remapped-value.mlir b/mlir/test/Transforms/test-legalize-remapped-value.mlir new file mode 100644 index 00000000000..ff571c93f93 --- /dev/null +++ b/mlir/test/Transforms/test-legalize-remapped-value.mlir @@ -0,0 +1,13 @@ +// RUN: mlir-opt %s -test-remapped-value | FileCheck %s + +// Simple test that exercises ConvertPatternRewriter::getRemappedValue. +func @remap_input_1_to_1(%arg0: i32) { + %0 = "test.one_variadic_out_one_variadic_in1"(%arg0) : (i32) -> i32 + %1 = "test.one_variadic_out_one_variadic_in1"(%0) : (i32) -> i32 + "test.return"() : () -> () +} +// CHECK-LABEL: func @remap_input_1_to_1 +// CHECK-SAME: (%[[ARG:.*]]: i32) +// CHECK-NEXT: %[[VAL:.*]] = "test.one_variadic_out_one_variadic_in1"(%[[ARG]], %[[ARG]]) +// CHECK-NEXT: "test.one_variadic_out_one_variadic_in1"(%[[VAL]], %[[VAL]]) + |
