From 7343b20dc89f3819f8e1f1d2bfd04a4c4aecce83 Mon Sep 17 00:00:00 2001 From: Kavya Srinet Date: Wed, 14 Feb 2018 14:53:24 -0800 Subject: [PATCH] Fixed op_registry_test.cc --- paddle/fluid/framework/op_registry_test.cc | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/paddle/fluid/framework/op_registry_test.cc b/paddle/fluid/framework/op_registry_test.cc index 02853ef878..0d791c8583 100644 --- a/paddle/fluid/framework/op_registry_test.cc +++ b/paddle/fluid/framework/op_registry_test.cc @@ -290,8 +290,7 @@ class OpWithMultiKernelTest : public OperatorWithKernel { framework::OpKernelType GetExpectedKernelType( const framework::ExecutionContext& ctx) const override { - return framework::OpKernelType(proto::VarType::FP32, - platform::CUDAPlace(0), + return framework::OpKernelType(proto::VarType::FP32, platform::CUDAPlace(0), DataLayout::kAnyLayout, framework::LibraryType::kCUDNN); } -- GitLab