You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@mxnet.apache.org by je...@apache.org on 2021/12/16 03:01:34 UTC
[incubator-mxnet] branch v1.x updated: Port changes from master to make CPP package properly build when large tensor support is enabled. (#20768)
This is an automated email from the ASF dual-hosted git repository.
jevans pushed a commit to branch v1.x
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
The following commit(s) were added to refs/heads/v1.x by this push:
new 60cab18 Port changes from master to make CPP package properly build when large tensor support is enabled. (#20768)
60cab18 is described below
commit 60cab18616ddfe2191797582a63f834b15e58ecb
Author: Joe Evans <jo...@gmail.com>
AuthorDate: Wed Dec 15 18:59:25 2021 -0800
Port changes from master to make CPP package properly build when large tensor support is enabled. (#20768)
---
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 853c519..8e28c89 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 = ''