Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[TOPI] Enable standalone wheel build #3657

Merged
merged 2 commits into from
Jul 30, 2019
Merged

[TOPI] Enable standalone wheel build #3657

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
d530bf6
Fixed topi bdist_wheel build to include libraries.
jwfromm Jul 29, 2019
fc64704
Removed unneeded imports
jwfromm Jul 29, 2019
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
55 changes: 49 additions & 6 deletions topi/python/setup.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,8 +18,9 @@
# pylint: disable=invalid-name, exec-used
"""Setup TOPI package."""
from __future__ import absolute_import
import sys
import os
import shutil
import sys

from setuptools import find_packages
from setuptools.dist import Distribution
Expand All @@ -31,21 +32,25 @@
from setuptools import setup
from setuptools.extension import Extension

CURRENT_DIR = os.path.dirname(__file__)


def get_lib_names():
if sys.platform.startswith('win32'):
return ['libtvm_topi.dll', 'tvm_topi.dll']
if sys.platform.startswith('darwin'):
return ['libtvm_topi.dylib', 'tvm_topi.dylib']
return ['libtvm_topi.so', 'tvm_topi.so']


def get_lib_path():
"""Get library path, name and version"""
# We can not import `libinfo.py` in setup.py directly since __init__.py
# Will be invoked which introduces dependences
CURRENT_DIR = os.path.dirname(__file__)
libinfo_py = os.path.join(CURRENT_DIR, '../../python/tvm/_ffi/libinfo.py')
libinfo = {'__file__': libinfo_py}
exec(compile(open(libinfo_py, "rb").read(), libinfo_py, 'exec'), libinfo, libinfo)
exec(compile(open(libinfo_py, "rb").read(),
libinfo_py, 'exec'), libinfo, libinfo)
version = libinfo['__version__']
if not os.getenv('CONDA_BUILD'):
lib_path = libinfo['find_lib_path'](get_lib_names())
Expand All @@ -59,6 +64,7 @@ def get_lib_path():
libs = None
return libs, version


LIB_LIST, __version__ = get_lib_path()

if not os.getenv('CONDA_BUILD'):
Expand All @@ -72,13 +78,50 @@ def get_lib_path():
else:
setup_kwargs = {}


include_libs = False
wheel_include_libs = False
if not os.getenv('CONDA_BUILD'):
if "bdist_wheel" in sys.argv:
wheel_include_libs = True
else:
include_libs = True

# For bdist_wheel only
if wheel_include_libs:
with open("MANIFEST.in", "w") as fo:
for path in LIB_LIST:
shutil.copy(path, os.path.join(CURRENT_DIR, 'topi'))
_, libname = os.path.split(path)
fo.write("include topi/%s\n" % libname)
setup_kwargs = {
"include_package_data": True
}

if include_libs:
curr_path = os.path.dirname(os.path.abspath(os.path.expanduser(__file__)))
for i, path in enumerate(LIB_LIST):
LIB_LIST[i] = os.path.relpath(path, curr_path)
setup_kwargs = {
"include_package_data": True,
"data_files": [('topi', LIB_LIST)]
}

setup(name='topi',
version=__version__,
description="TOPI: TVM operator index",
install_requires=[
"numpy",
"decorator",
],
"numpy",
"decorator",
],
packages=find_packages(),
url='https://github.com/dmlc/tvm',
**setup_kwargs)


if wheel_include_libs:
# Wheel cleanup
os.remove("MANIFEST.in")
for path in LIB_LIST:
_, libname = os.path.split(path)
os.remove("topi/%s" % libname)