From 08514eaa5f59a8a29f01d04ec1bb7f0e41bb0fea Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Wed, 23 Sep 2020 05:50:02 -0700
Subject: [PATCH] [MLIR][KernelGen] Add E2E tests for `tf.Tan` and `tf.Sinh`

PiperOrigin-RevId: 333275089
---
 include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops.td | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops.td b/include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops.td
index 9225d02..6e6af20 100644
--- a/include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops.td
+++ b/include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops.td
@@ -477,7 +477,8 @@ def ReshapeMemRefCastOp: Op<LHLO_Dialect, "reshape_memref_cast", [
   let results = (outs AnyRankedOrUnrankedMemRef:$result);
 
   let extraClassDeclaration = [{
-    MemRefType getType() { return getResult().getType().cast<MemRefType>(); }
+    BaseMemRefType getType() {
+        return getResult().getType().cast<BaseMemRefType>(); }
   }];
 
   let verifier = [{ return Verify(*this); }];