forked from microsoft/nn-Meter
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
43 lines (41 loc) · 1.73 KB
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
from setuptools import setup, find_packages
setup(
name='nn-meter',
version='2.0dev',
description='nn-Meter is a novel and efficient system to accurately predict the inference latency of DNN models on diverse edge devices.',
long_description = open('README.md', encoding='utf-8').read(),
long_description_content_type = 'text/markdown',
author='Microsoft nn-Meter Team',
author_email='nn-meter-dev@microsoft.com',
url='https://github.com/microsoft/nn-Meter',
project_urls={
'Data of models': 'https://github.com/microsoft/nn-Meter/releases/tag/v1.0-data',
},
license = 'MIT',
classifiers = [
'License :: OSI Approved :: MIT License',
'Operating System :: MacOS :: MacOS X',
'Operating System :: Microsoft :: Windows :: Windows 10',
'Operating System :: POSIX :: Linux',
'Programming Language :: Python :: 3 :: Only',
'Topic :: Scientific/Engineering :: Artificial Intelligence',
],
packages=find_packages(),
package_data={
'nn_meter': [
'configs/*.yaml',
'configs/builder/backends/*.yaml',
'configs/builder/fusion_rule_tester/*.yaml',
'configs/builder/predictor_builder/*.yaml',
'builder/kernel_predictor_builder/data_sampler/prior_config_lib/*.csv',
'kernel_detector/fusion_lib/*.json'],
},
entry_points={
'console_scripts': ['nn-meter=nn_meter.utils.nn_meter_cli.interface:nn_meter_cli'],
},
install_requires=[
'numpy', 'pandas', 'tqdm', 'networkx', 'requests', 'protobuf', 'PyYAML', 'scikit_learn', 'packaging', 'jsonlines'
],
)