From 9a3c6d5f9c50ba170337881953890c41fa9cf0f3 Mon Sep 17 00:00:00 2001
From: fengjiayi <fengjiayi@baidu.com>
Date: Wed, 20 Sep 2017 17:56:08 -0700
Subject: [PATCH] Add BlockDesc and ProgramDesc to framework.proto

---
 paddle/framework/framework.proto | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/paddle/framework/framework.proto b/paddle/framework/framework.proto
index dfcb5fb6210..dcbc9ec4076 100644
--- a/paddle/framework/framework.proto
+++ b/paddle/framework/framework.proto
@@ -23,6 +23,7 @@ enum AttrType {
   FLOATS = 4;
   STRINGS = 5;
   INT_PAIRS = 6;
+  BLOCK = 7;
 }
 
 message IntPair {
@@ -44,6 +45,7 @@ message OpDesc {
     repeated float floats = 7;
     repeated string strings = 8;
     repeated IntPair int_pairs = 9;
+    optional int32 block = 10;
   };
 
   message Var {
@@ -108,3 +110,12 @@ message VarDesc {
   required string name = 1;
   optional LoDTensorDesc lod_tensor = 2;
 }
+
+message BlockDesc {
+  required int32 idx = 1;
+  required int32 parent_idx = 2;
+  repeated VarDesc vars = 3;
+  repeated OpDesc ops = 4;
+}
+
+message ProgramDesc { repeated BlockDesc blocks = 1; }
-- 
GitLab