Fix macro

apache · Sep 25, 2020 · 9101520 · 9101520
1 parent 05f581d
commit 9101520
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/src/operator/nn/concat.cc b/src/operator/nn/concat.cc
@@ -326,6 +326,7 @@ DMLC_REGISTER_PARAMETER(ConcatParam);
     [](const NodeAttrs& attrs) { \
   const ConcatParam& params = nnvm::get<ConcatParam>(attrs.parsed); \
   std::vector<std::string> ret; \
+  ret.reserve(params.num_args); \
   for (int i = 0; i < params.num_args; ++i) { \
     ret.push_back(std::string("arg") + std::to_string(i)); \
   } \

diff --git a/src/operator/numpy/np_init_op.cc b/src/operator/numpy/np_init_op.cc
@@ -161,6 +161,7 @@ NNVM_REGISTER_OP(_npi_atleast_##N##d)                                      \
 [](const nnvm::NodeAttrs& attrs) {                                        \
   int num_args = nnvm::get<AtleastNDParam>(attrs.parsed).num_args;        \
   std::vector<std::string> ret;                                           \
+  ret.reserve(num_args);                                                  \
   for (int i = 0; i < num_args; i++) {                                    \
     ret.push_back(std::string("ary") + std::to_string(i));                \
   }                                                                       \