From 5252aeae8f706890e4461abf8fcacb984f4a6bd7 Mon Sep 17 00:00:00 2001
From: Robert Suderman <suderman@google.com>
Date: Wed, 12 Aug 2020 16:14:42 -0700
Subject: [PATCH] Lowering for mhlo.ceil to std.ceil

PiperOrigin-RevId: 326335301
---
 .../mhlo/transforms/legalize_to_standard_patterns.td     | 4 ++++
 tests/legalize-to-std.mlir                               | 9 +++++++++
 2 files changed, 13 insertions(+)

diff --git a/lib/Dialect/mhlo/transforms/legalize_to_standard_patterns.td b/lib/Dialect/mhlo/transforms/legalize_to_standard_patterns.td
index ea67c05..6ee6f12 100644
--- a/lib/Dialect/mhlo/transforms/legalize_to_standard_patterns.td
+++ b/lib/Dialect/mhlo/transforms/legalize_to_standard_patterns.td
@@ -36,6 +36,10 @@ def IsSameSizePred : CPred<
 def IsSameSizeConstraint : Constraint<IsSameSizePred, "inputs are same size">;
 
 
+// Unary Lowering Patterns.
+def : Pat<(HLO_CeilOp HLO_FpTensor:$i), (CeilFOp $i)>;
+
+// Binary Lowering Patterns.
 def : Pat<(HLO_AndOp HLO_PredTensor:$l, HLO_PredTensor:$r),
           (AndOp $l, $r),
           [(IsSameSizeConstraint $l, $r)]>;
diff --git a/tests/legalize-to-std.mlir b/tests/legalize-to-std.mlir
index 37a6149..abe4e87 100644
--- a/tests/legalize-to-std.mlir
+++ b/tests/legalize-to-std.mlir
@@ -42,6 +42,15 @@ func @binary_ops_int(%arg0: tensor<4xi32>, %arg1: tensor<4xi32>) -> tensor<4xi32
   return %4 : tensor<4xi32>
 }
 
+// CHECK-LABEL: func @unary_ops_float
+func @unary_ops_float(%arg0: tensor<4xf32>) -> tensor<4xf32> {
+  // CHECK-NEXT: %0 = ceilf %arg0 : tensor<4xf32>
+  %0 = "mhlo.ceil"(%arg0) : (tensor<4xf32>) -> tensor<4xf32>
+
+  // CHECK-NEXT:   return %0 : tensor<4xf32>
+  return %0 : tensor<4xf32>
+}
+
 // CHECK-LABEL: func @compare_int(%arg0: tensor<4xi32>) -> (tensor<4xi1>, tensor<4xi1>, tensor<4xi1>, tensor<4xi1>, tensor<4xi1>, tensor<4xi1>) {
 func @compare_int(%arg0: tensor<4xi32>) -> (tensor<4xi1>,tensor<4xi1>,tensor<4xi1>,tensor<4xi1>,tensor<4xi1>,tensor<4xi1>) {
   // CHECK-NEXT: %0 = cmpi "eq", %arg0, %arg0 : tensor<4xi32>