added copying lib_api.cc into pip wheel for building extensions

apache · Oct 21, 2020 · 3d28436 · 3d28436
1 parent ed11064
commit 3d28436
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tools/pip/setup.py b/tools/pip/setup.py
@@ -101,6 +101,8 @@
                 os.path.join(CURRENT_DIR, 'mxnet/include/mshadow'))
 shutil.copytree(os.path.join(CURRENT_DIR, 'mxnet-build/3rdparty/tvm/nnvm/include/nnvm'),
                 os.path.join(CURRENT_DIR, 'mxnet/include/nnvm'))
+shutil.copytree(os.path.join(CURRENT_DIR, 'mxnet/src/lib_api.cc'),
+                os.path.join(CURRENT_DIR, 'mxnet/src'))
 
 package_name = 'mxnet'