Commit 31b47c84 by Wei Chen Committed by Tianqi Chen

[Relay] Install Relay Prelude program in package install (#4227)

parent 83385d42
...@@ -138,6 +138,10 @@ if include_libs: ...@@ -138,6 +138,10 @@ if include_libs:
"data_files": [('tvm', LIB_LIST)] "data_files": [('tvm', LIB_LIST)]
} }
def get_package_data_files():
# Relay standard libraries
return ['relay/std/prelude.rly']
setup(name='tvm', setup(name='tvm',
version=__version__, version=__version__,
description="TVM: An End to End Tensor IR/DSL Stack for Deep Learning Systems", description="TVM: An End to End Tensor IR/DSL Stack for Deep Learning Systems",
...@@ -149,6 +153,8 @@ setup(name='tvm', ...@@ -149,6 +153,8 @@ setup(name='tvm',
'psutil', 'psutil',
], ],
packages=find_packages(), packages=find_packages(),
package_dir={'tvm': 'tvm'},
package_data={'tvm': get_package_data_files()},
distclass=BinaryDistribution, distclass=BinaryDistribution,
url='https://github.com/dmlc/tvm', url='https://github.com/dmlc/tvm',
ext_modules=config_cython(), ext_modules=config_cython(),
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment