From 33c0c8e3cf1fec3abd4c60fd0f2f4427843810e0 Mon Sep 17 00:00:00 2001
From: Joe Evans <joseph.evans@gmail.com>
Date: Mon, 24 Jan 2022 18:11:22 -0800
Subject: [PATCH] Port changes from master to make CPP package properly build
 when large tensor support is enabled. (#20768) (#20841)

---
 cpp-package/scripts/OpWrapperGenerator.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/cpp-package/scripts/OpWrapperGenerator.py b/cpp-package/scripts/OpWrapperGenerator.py
index c09b156758d1..9b08e6b48704 100644
--- 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 = ''