colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
mlir-hlo/tests/hlo-legalize-to-lhlo-unrank...

45 lines
1.7 KiB
MLIR
Raw Permalink Normal View History

PR #49454: [MLIR][DISC] Upgrade to use the new `reifyReturnTypeShapes` interface. Imported from GitHub PR https://github.com/tensorflow/tensorflow/pull/49454 The new interface is more safe to be used during dialect conversion (e.g. converting from tensor world to buffer world). Copybara import of the project: -- a6968072d59bec3c3bbaef0121d297e807c37c91 by Wenyi Zhao <reyizero@gmail.com>: [MLIR][DISC] Upgrade to use the new `reifyReturnTypeShapes` interface. The new interface is more safe to be used during dialect conversion (e.g. converting from tensor world to buffer world). -- 55e7c6b7f2f99b99e226645a57e2433fae3e90ed by Wenyi Zhao <reyizero@gmail.com>: minor fix PiperOrigin-RevId: 375500273
2021-05-25 01:10:21 +08:00
// RUN: mlir-hlo-opt -hlo-legalize-to-lhlo -buffer-hoisting -buffer-deallocation -canonicalize %s -o - | FileCheck %s
[MLIR] Convert FuncOp signature with unranked types in HLO->LHLO conversion. PiperOrigin-RevId: 320146856
2020-07-08 16:43:30 +08:00
// CHECK-LABEL: func @func_op_unranked_arg_result
func @func_op_unranked_arg_result(%arg0: tensor<*xf32>) -> tensor<*xf32> {
return %arg0 : tensor<*xf32>
}
// CHECK-SAME: ([[ARG:%.*]]: memref<*xf32>) -> memref<*xf32>
// CHECK-NEXT: return [[ARG]] : memref<*xf32>
[MLIR][LHLO] Convert mhlo.dynamic_reshape -> lhlo.reshape_memref_cast. PiperOrigin-RevId: 320149593
2020-07-08 17:11:30 +08:00
// -----
// CHECK-LABEL: func @dynamic_reshape_from_unranked
func @dynamic_reshape_from_unranked(
%operand: tensor<*xf32>, %shape: tensor<1xi32>) -> tensor<?xf32> {
%reshaped = "mhlo.dynamic_reshape"(%operand, %shape)
: (tensor<*xf32>, tensor<1xi32>) -> tensor<?xf32>
return %reshaped : tensor<?xf32>
}
// CHECK-SAME: ([[ARG:%.*]]: memref<*xf32>, [[SHAPE:%.*]]: memref<1xi32>)
Integrate LLVM at llvm/llvm-project@678241795c95 Updates LLVM usage to match [678241795c95](https://github.com/llvm/llvm-project/commit/678241795c95) PiperOrigin-RevId: 363257913
2021-03-17 04:31:59 +08:00
// CHECK-NEXT: memref.reshape [[ARG]]([[SHAPE]])
[MLIR][LHLO] Convert mhlo.dynamic_reshape -> lhlo.reshape_memref_cast. PiperOrigin-RevId: 320149593
2020-07-08 17:11:30 +08:00
// CHECK-SAME: : (memref<*xf32>, memref<1xi32>) -> memref<?xf32>
// -----
// CHECK-LABEL: func @dynamic_reshape_to_unranked
func @dynamic_reshape_to_unranked(
%operand: tensor<?xf32>, %shape: tensor<?xi32>) -> tensor<*xf32> {
%reshaped = "mhlo.dynamic_reshape"(%operand, %shape)
: (tensor<?xf32>, tensor<?xi32>) -> tensor<*xf32>
return %reshaped : tensor<*xf32>
}
// CHECK-SAME: ([[ARG:%.*]]: memref<?xf32>, [[SHAPE:%.*]]: memref<?xi32>)
Integrate LLVM at llvm/llvm-project@678241795c95 Updates LLVM usage to match [678241795c95](https://github.com/llvm/llvm-project/commit/678241795c95) PiperOrigin-RevId: 363257913
2021-03-17 04:31:59 +08:00
// CHECK-NEXT: memref.reshape [[ARG]]([[SHAPE]])
[MLIR][LHLO] Convert mhlo.dynamic_reshape -> lhlo.reshape_memref_cast. PiperOrigin-RevId: 320149593
2020-07-08 17:11:30 +08:00
// CHECK-SAME: : (memref<?xf32>, memref<?xi32>) -> memref<*xf32>
[KERNEL_GEN] Add a pattern to bufferize `mhlo.reshape(<unranked_tensor>)`. PiperOrigin-RevId: 356720899
2021-02-10 22:30:50 +08:00
// -----
// CHECK-LABEL: func @reshape_unranked
func @reshape_unranked(%operand: tensor<*xf32>) -> tensor<f32> {
%reshaped = "mhlo.reshape"(%operand) : (tensor<*xf32>) -> tensor<f32>
return %reshaped : tensor<f32>
}
// CHECK-SAME: ([[ARG:%.*]]: memref<*xf32>)
Integrate LLVM at llvm/llvm-project@678241795c95 Updates LLVM usage to match [678241795c95](https://github.com/llvm/llvm-project/commit/678241795c95) PiperOrigin-RevId: 363257913
2021-03-17 04:31:59 +08:00
// CHECK-NEXT: memref.cast [[ARG]] : memref<*xf32> to memref<f32>