From 76b18768c3cfa833e8c178abd4ade4eb09796f23 Mon Sep 17 00:00:00 2001 From: Roman Dzhabarov Date: Thu, 12 Nov 2020 09:59:21 -0800 Subject: [PATCH] Fixing Windows build. PiperOrigin-RevId: 342070912 Change-Id: I293a6865466ca5cbddc2d7d18e451b78a36fd9d7 --- .../mlir/mlir_graph_optimization_pass_test.cc | 15 +++++++-------- 1 file changed, 7 insertions(+), 8 deletions(-) diff --git a/tensorflow/compiler/mlir/mlir_graph_optimization_pass_test.cc b/tensorflow/compiler/mlir/mlir_graph_optimization_pass_test.cc index a7372437fdf..799f0562baf 100644 --- a/tensorflow/compiler/mlir/mlir_graph_optimization_pass_test.cc +++ b/tensorflow/compiler/mlir/mlir_graph_optimization_pass_test.cc @@ -28,14 +28,13 @@ using ::testing::Test; class MockMlirOptimizationPass : public MlirOptimizationPass { public: - MOCK_METHOD(llvm::StringRef, name, (), (const, override)); - MOCK_METHOD(bool, IsEnabled, - (const ConfigProto& config_proto, const Graph& graph), - (const, override)); - MOCK_METHOD(Status, Run, - (const ConfigProto& config_proto, mlir::ModuleOp module, - const Graph& graph), - (override)); + // MOCK_METHOD does not work on Windows build, using MOCK_METHODX + // instead. + MOCK_CONST_METHOD0(name, llvm::StringRef()); + MOCK_CONST_METHOD2(IsEnabled, + bool(const ConfigProto& config_proto, const Graph& graph)); + MOCK_METHOD3(Run, Status(const ConfigProto& config_proto, + mlir::ModuleOp module, const Graph& graph)); }; class MlirGraphOptimizationPassTest : public Test {