From 3da6c853400ab19557108b865f493f8ec73fb538 Mon Sep 17 00:00:00 2001
From: Helin Wang <helinwang@baidu.com>
Date: Thu, 11 May 2017 17:40:51 -0700
Subject: [PATCH] use enum instead of define

---
 doc/design/cluster_train/pserver_client.md | 22 ++++++++++++----------
 1 file changed, 12 insertions(+), 10 deletions(-)

diff --git a/doc/design/cluster_train/pserver_client.md b/doc/design/cluster_train/pserver_client.md
index e1fa4526396..0a45a861177 100644
--- a/doc/design/cluster_train/pserver_client.md
+++ b/doc/design/cluster_train/pserver_client.md
@@ -58,18 +58,20 @@ The selected trainer's call to `paddle_begin_init_params` will return with 1, an
 ## C Interface
 
 ```c
-#define PADDLE_ELEMENT_TYPE_INT32   0
-#define PADDLE_ELEMENT_TYPE_UINT32  1
-#define PADDLE_ELEMENT_TYPE_INT64   2
-#define PADDLE_ELEMENT_TYPE_UINT64  3
-#define PADDLE_ELEMENT_TYPE_FLOAT32 4
-#define PADDLE_ELEMENT_TYPE_FLOAT64 5
+typedef enum {
+  PADDLE_ELEMENT_TYPE_INT32   = 0,
+  PADDLE_ELEMENT_TYPE_UINT32  = 1,
+  PADDLE_ELEMENT_TYPE_INT64   = 2,
+  PADDLE_ELEMENT_TYPE_UINT64  = 3,
+  PADDLE_ELEMENT_TYPE_FLOAT32 = 4,
+  PADDLE_ELEMENT_TYPE_FLOAT64 = 5,
+} paddle_element_type;
 
 typedef struct {
-  char* name;
-  int   element_type;
-  void* content;
-  int   content_len;
+  char*               name;
+  paddle_element_type element_type;
+  void*               content;
+  int                 content_len;
 } paddle_parameter, paddle_gradient;
 
 typedef struct paddle_pserver_client paddle_pserver_client;
-- 
GitLab