File tree Expand file tree Collapse file tree 3 files changed +5
-2
lines changed Expand file tree Collapse file tree 3 files changed +5
-2
lines changed Original file line number Diff line number Diff line change @@ -27,6 +27,7 @@ enum AttrType {
27
27
BOOLEANS = 7 ;
28
28
BLOCK = 8 ;
29
29
LONG = 9 ;
30
+ BLOCKS = 10 ;
30
31
}
31
32
32
33
// OpDesc describes an instance of a C++ framework::OperatorBase
Original file line number Diff line number Diff line change @@ -342,7 +342,8 @@ class ListenAndServOpMaker : public framework::OpProtoAndCheckerMaker {
342
342
.SetDefault ({});
343
343
AddAttr<bool >(" sync_mode" , " if works at sync_mode or not" ).SetDefault (true );
344
344
AddAttr<std::vector<framework::BlockDesc *>>(
345
- kOptimizeBlocks , " Optimize blocks to run on server side." );
345
+ kOptimizeBlocks , " Optimize blocks to run on server side." )
346
+ .SetDefault ({});
346
347
AddAttr<std::vector<std::string>>(kPrefetchVarNameToBlockId ,
347
348
" prefetch blocks to run on server side." )
348
349
.SetDefault ({});
Original file line number Diff line number Diff line change @@ -268,7 +268,8 @@ void BindOpDesc(pybind11::module *m) {
268
268
.value (" STRINGS" , pd::proto::AttrType::STRINGS)
269
269
.value (" BOOL" , pd::proto::AttrType::BOOLEAN)
270
270
.value (" BOOLS" , pd::proto::AttrType::BOOLEANS)
271
- .value (" BLOCK" , pd::proto::AttrType::BLOCK);
271
+ .value (" BLOCK" , pd::proto::AttrType::BLOCK)
272
+ .value (" BLOCKS" , pd::proto::AttrType::BLOCKS);
272
273
273
274
pybind11::class_<pd::OpDesc> op_desc (*m, " OpDesc" , " " );
274
275
op_desc
You can’t perform that action at this time.
0 commit comments