Port changes from master to make CPP package properly build when larg…

…e tensor support is enabled. (apache#20768) (apache#20841)
zmughal · Jan 25, 2022 · 33c0c8e · 33c0c8e
1 parent 0642923
commit 33c0c8e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/cpp-package/scripts/OpWrapperGenerator.py b/cpp-package/scripts/OpWrapperGenerator.py
@@ -99,7 +99,9 @@ class Arg:
         'double or None':'dmlc::optional<double>',\
         'Shape or None':'dmlc::optional<Shape>',\
         'string':'const std::string&',\
-        'tuple of <float>':'nnvm::Tuple<mx_float>'}
+        'tuple of <float>':'nnvm::Tuple<mx_float>',\
+        'tuple of <>':'mxnet::cpp::Shape',\
+        '':'index_t'}
     name = ''
     type = ''
     description = ''