[PIP] update manifest to include lib_api.cc (#19850)

* update manifest to include lib_api.cc Signed-off-by: Sheng Zha <zhasheng@amazon.com> * add cc Signed-off-by: Sheng Zha <zhasheng@amazon.com> * fix lib_api.cc packaging Signed-off-by: Sheng Zha <zhasheng@amazon.com>
apache · Feb 17, 2021 · 9ef0fe9 · 9ef0fe9
1 parent cda168b
commit 9ef0fe9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/tools/pip/MANIFEST.in b/tools/pip/MANIFEST.in
@@ -29,4 +29,6 @@ recursive-include mxnet *_LICENSE
 recursive-include mxnet *.h
 recursive-include mxnet *.hpp
 recursive-include mxnet *.cuh
+recursive-include mxnet *.cc
+recursive-include mxnet *.cpp
 recursive-include dmlc_tracker *.py
diff --git a/tools/pip/setup.py b/tools/pip/setup.py
@@ -105,6 +105,7 @@ def has_ext_modules(self):
                 os.path.join(CURRENT_DIR, 'mxnet/include/nnvm'))
 
 # copy cc file for mxnet extensions
+os.mkdir(os.path.join(CURRENT_DIR, 'mxnet/src'))
 shutil.copy(os.path.join(CURRENT_DIR, 'mxnet-build/src/lib_api.cc'),
             os.path.join(CURRENT_DIR, 'mxnet/src'))