From 15700964e17de3ed0ca4d847976796937acf4932 Mon Sep 17 00:00:00 2001 From: Sheng Zha Date: Sun, 7 Feb 2021 16:00:37 -0500 Subject: [PATCH] [PIP] update manifest to include lib_api.cc (#19850) * update manifest to include lib_api.cc Signed-off-by: Sheng Zha * add cc Signed-off-by: Sheng Zha * fix lib_api.cc packaging Signed-off-by: Sheng Zha --- tools/pip/MANIFEST.in | 2 ++ tools/pip/setup.py | 1 + 2 files changed, 3 insertions(+) diff --git a/tools/pip/MANIFEST.in b/tools/pip/MANIFEST.in index 0b042e303097..2718efdf7e6b 100644 --- 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 index a5af13d8f5eb..fda7afa48ac8 100644 --- 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'))