Skip to content

Commit 3415e26

Browse files
committed
Remove some layers in mobile inference library.
1 parent 2368377 commit 3415e26

File tree

4 files changed

+48
-4
lines changed

4 files changed

+48
-4
lines changed

paddle/cuda/CMakeLists.txt

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,9 @@ if(WITH_GPU)
2727
set_source_files_properties(${CUDA_CXX_SOURCES}
2828
PROPERTIES COMPILE_FLAGS "-D__NVCC__")
2929
else()
30+
if (NOT MOBILE_INFERENCE)
3031
set(CUDA_CXX_SOURCES src/hl_warpctc_wrap.cc)
32+
endif()
3133
endif()
3234

3335
set(CUDA_CU_SOURCES

paddle/gserver/CMakeLists.txt

Lines changed: 42 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -85,9 +85,49 @@ if(MOBILE_INFERENCE)
8585
gradientmachines/GradientMachineMode.cpp
8686
gradientmachines/MultiGradientMachine.cpp)
8787

88-
# Remove useless layers
88+
# Remove layers that used in training
8989
list(REMOVE_ITEM GSERVER_SOURCES
90-
layers/RecurrentLayerGroup.cpp)
90+
layers/RecurrentLayerGroup.cpp
91+
layers/CostLayer.cpp
92+
layers/MultiBoxLossLayer.cpp
93+
layers/WarpCTCLayer.cpp
94+
layers/CTCLayer.cpp
95+
layers/LinearChainCTC.cpp
96+
layers/PrintLayer.cpp)
97+
list(REMOVE_ITEM GSERVER_SOURCES
98+
layers/OuterProdLayer.cpp
99+
layers/SumToOneNormLayer.cpp
100+
layers/ConvShiftLayer.cpp
101+
layers/InterpolationLayer.cpp
102+
layers/AgentLayer.cpp
103+
layers/DotMulOperator.cpp
104+
layers/GruStepLayer.cpp
105+
layers/LstmStepLayer.cpp
106+
layers/ConvexCombinationLayer.cpp
107+
layers/Conv3DLayer.cpp
108+
layers/DeConv3DLayer.cpp
109+
layers/CropLayer.cpp
110+
layers/CrossEntropyOverBeam.cpp
111+
layers/DataNormLayer.cpp
112+
layers/FeatureMapExpandLayer.cpp
113+
layers/HierarchicalSigmoidLayer.cpp
114+
layers/MultinomialSampler.cpp
115+
layers/NCELayer.cpp
116+
layers/KmaxSeqScoreLayer.cpp
117+
layers/MDLstmLayer.cpp
118+
layers/MultiplexLayer.cpp
119+
layers/PadLayer.cpp
120+
layers/Pool3DLayer.cpp
121+
layers/ResizeLayer.cpp
122+
layers/RotateLayer.cpp
123+
layers/RowConvLayer.cpp
124+
layers/RowL2NormLayer.cpp
125+
layers/SamplingIdLayer.cpp
126+
layers/ScaleShiftLayer.cpp
127+
layers/SelectiveFullyConnectedLayer.cpp
128+
layers/SpatialPyramidPoolLayer.cpp
129+
layers/BilinearInterpLayer.cpp
130+
layers/ClipLayer.cpp)
91131
endif()
92132

93133
if(WITH_GPU)

paddle/gserver/layers/Layer.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -98,6 +98,7 @@ ClassRegistrar<Layer, LayerConfig> Layer::registrar_;
9898
LayerPtr Layer::create(const LayerConfig& config) {
9999
std::string type = config.type();
100100

101+
#ifndef PADDLE_MOBILE_INFERENCE
101102
// NOTE: As following types have illegal character '-',
102103
// they can not use REGISTER_LAYER to registrar.
103104
// Besides, to fit with old training models,
@@ -106,7 +107,6 @@ LayerPtr Layer::create(const LayerConfig& config) {
106107
return LayerPtr(new MultiClassCrossEntropy(config));
107108
else if (type == "rank-cost")
108109
return LayerPtr(new RankingCost(config));
109-
#ifndef PADDLE_MOBILE_INFERENCE
110110
else if (type == "auc-validation")
111111
return LayerPtr(new AucValidation(config));
112112
else if (type == "pnpair-validation")

paddle/gserver/tests/CMakeLists.txt

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -112,8 +112,10 @@ endif()
112112
################ test_LinearChainCRF ####################
113113
add_simple_unittest(test_LinearChainCRF)
114114

115+
if(NOT MOBILE_INFERENCE)
115116
############## test_MultinomialSampler ###################
116117
add_simple_unittest(test_MultinomialSampler)
118+
endif()
117119

118120
############## test_PyDataProvider ########################
119121
if(WITH_PYTHON)
@@ -129,7 +131,7 @@ endif()
129131
add_simple_unittest(test_RecurrentLayer)
130132

131133
############### test_WarpCTCLayer #######################
132-
if(NOT WITH_DOUBLE)
134+
if(NOT WITH_DOUBLE AND NOT MOBILE_INFERENCE)
133135
add_unittest_without_exec(test_WarpCTCLayer
134136
test_WarpCTCLayer.cpp)
135137

0 commit comments

Comments
 (0)