From ec1d63af5683d15c829d8f43ffaab8ab0c1b7bf4 Mon Sep 17 00:00:00 2001
From: Kavya Srinet <kavyasrinet@baidu.com>
Date: Wed, 14 Feb 2018 11:23:10 -0800
Subject: [PATCH] fixed protobuf.cc

---
 paddle/fluid/pybind/protobuf.cc | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

diff --git a/paddle/fluid/pybind/protobuf.cc b/paddle/fluid/pybind/protobuf.cc
index 9f97cc5007e..1b8fc1698b9 100644
--- a/paddle/fluid/pybind/protobuf.cc
+++ b/paddle/fluid/pybind/protobuf.cc
@@ -195,14 +195,14 @@ void BindBlockDesc(py::module &m) {
 }
 
 void BindVarDsec(py::module &m) {
-  py::enum_<proto::DataType>(m, "DataType", "")
-      .value("BOOL", proto::DataType::BOOL)
-      .value("INT16", proto::DataType::INT16)
-      .value("INT32", proto::DataType::INT32)
-      .value("INT64", proto::DataType::INT64)
-      .value("FP16", proto::DataType::FP16)
-      .value("FP32", proto::DataType::FP32)
-      .value("FP64", proto::DataType::FP64);
+  py::enum_<proto::VarType::Type>(m, "DataType", "")
+      .value("BOOL", proto::VarType::Type::BOOL)
+      .value("INT16", proto::VarType::Type::INT16)
+      .value("INT32", proto::VarType::Type::INT32)
+      .value("INT64", proto::VarType::Type::INT64)
+      .value("FP16", proto::VarType::Type::FP16)
+      .value("FP32", proto::VarType::Type::FP32)
+      .value("FP64", proto::VarType::Type::FP64);
 
   py::class_<VarDesc> var_desc(m, "VarDesc", "");
   var_desc
-- 
GitLab