modify all proto used in inference with RUNTIME_LITE, delete Unnecess…

…ary proto druning inference process
QingshuChen · Oct 9, 2017 · bc9d8b5 · bc9d8b5
1 parent 3a68955
commit bc9d8b5
Show file tree

Hide file tree

Showing 6 changed files with 16 additions and 1 deletion.
diff --git a/proto/CMakeLists.txt b/proto/CMakeLists.txt
@@ -1,4 +1,10 @@
-file(GLOB proto_filenames . *.proto)
+if (MOBILE_INFERENCE)
+    file(GLOB proto_filenames . ModelConfig.proto ParameterConfig.proto
+         TrainerConfig.proto DataConfig.proto)
+else()
+    file(GLOB proto_filenames . *.proto)
+endif()
+
 include_directories(${CMAKE_CURRENT_BINARY_DIR})
 proto_library(paddle_proto SRCS ${proto_filenames})
 

diff --git a/proto/DataConfig.proto b/proto/DataConfig.proto
@@ -13,6 +13,8 @@ See the License for the specific language governing permissions and
 limitations under the License. */
 syntax = "proto2";
 
+option optimize_for = LITE_RUNTIME;
+
 package paddle;
 
 message FileGroupConf {

diff --git a/proto/ModelConfig.proto b/proto/ModelConfig.proto
@@ -13,6 +13,7 @@ See the License for the specific language governing permissions and
 limitations under the License. */
 syntax = "proto2";
 
+option optimize_for = LITE_RUNTIME;
 import "ParameterConfig.proto";
 
 package paddle;

diff --git a/proto/ParameterConfig.proto b/proto/ParameterConfig.proto
@@ -13,6 +13,8 @@ See the License for the specific language governing permissions and
 limitations under the License. */
 syntax = "proto2";
 
+option optimize_for = LITE_RUNTIME;
+
 package paddle;
 
 /**

diff --git a/proto/ParameterService.proto b/proto/ParameterService.proto
@@ -13,6 +13,8 @@ See the License for the specific language governing permissions and
 limitations under the License. */
 syntax = "proto2";
 
+option optimize_for = LITE_RUNTIME;
+
 import "ParameterConfig.proto";
 import "TrainerConfig.proto";
 

diff --git a/proto/TrainerConfig.proto b/proto/TrainerConfig.proto
@@ -13,6 +13,8 @@ See the License for the specific language governing permissions and
 limitations under the License. */
 syntax = "proto2";
 
+option optimize_for = LITE_RUNTIME;
+
 import "DataConfig.proto";
 import "ModelConfig.proto";