Fix for test_lans failure (#20036)

Make NumTensors function unique to multi_lans.cc & multi_lamb.cc.
apache · Mar 16, 2021 · b8dc162 · b8dc162
1 parent aa92db0
commit b8dc162
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/operator/contrib/multi_lamb.cc b/src/operator/contrib/multi_lamb.cc
@@ -167,7 +167,7 @@ std::vector<std::string> LAMBParamToVector(uint32_t num_tensors,
   return ret;
 }
 
-inline uint32_t NumTensors(const nnvm::NodeAttrs& attrs) {
+static inline uint32_t NumTensors(const nnvm::NodeAttrs& attrs) {
   return static_cast<uint32_t>(dmlc::get<MultiLAMBParam>(attrs.parsed).num_tensors);
 }
 

diff --git a/src/operator/contrib/multi_lans.cc b/src/operator/contrib/multi_lans.cc
@@ -183,7 +183,7 @@ std::vector<std::string> LANSParamToVector(uint32_t num_tensors,
   return ret;
 }
 
-inline uint32_t NumTensors(const nnvm::NodeAttrs& attrs) {
+static inline uint32_t NumTensors(const nnvm::NodeAttrs& attrs) {
   return static_cast<uint32_t>(dmlc::get<MultiLANSParam>(attrs.parsed).num_tensors);
 }