66 Star 277 Fork 135

MindSpore/akg

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
setup.py 2.79 KB
一键复制 编辑 原始数据 按行查看 历史
# Copyright 2021-2022 Huawei Technologies Co., Ltd
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import os
from setuptools import setup
pwd = os.path.dirname(os.path.realpath(__file__))
def _read_file(filename):
with open(os.path.join(pwd, filename), encoding='UTF-8') as f:
return f.read()
version = _read_file('version.txt').replace("\n", "")
package_data = {
'akg': [
'*.so*',
'*.cuh',
'lib/*.so*',
'config/*',
'include/*',
'include/*/*',
'include/*/*/*',
'include/*/*/*/*'
]
}
def collect_dirs(cur_dir, dirs):
"""Collect all sub dirs in current dir."""
for root, all_dirs, _ in os.walk(cur_dir, followlinks=True):
for d in all_dirs:
if '.' in d:
continue
full_path = os.path.join(root, d)
package = full_path.replace(os.path.sep, '.')
dirs.append(package)
def find_files(where=None):
"""
Return a package list
'where' is the root directory list
"""
if where is None:
where = ['.']
dirs = [path.replace(os.path.sep, '.') for path in where]
for selected_root in where:
collect_dirs(selected_root, dirs)
packages = []
for d in dirs:
if d.endswith("__pycache__"):
continue
elif d.startswith("build."):
packages.append(d[6:])
else:
packages.append(d)
return packages
setup(
name='akg',
version=version,
author='The MindSpore Authors',
author_email='[email protected]',
url='https://www.mindspore.cn/',
download_url='https://gitee.com/mindspore/akg/tags',
project_urls={
'Sources': 'https://gitee.com/mindspore/akg',
'Issue Tracker': 'https://gitee.com/mindspore/akg/issues',
},
description="An optimizer for operators in Deep Learning Networks, which provides the ability to automatically "
"fuse ops with specific patterns.",
license='Apache 2.0',
package_data=package_data,
packages=find_files(['build/akg']),
package_dir={'akg': 'build/akg'},
include_package_data=True,
install_requires=[
'scipy >= 1.5.2',
'numpy >= 1.17.0',
'decorator >= 4.4.0'
],
classifiers=[
'License :: OSI Approved :: Apache Software License'
]
)
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/mindspore/akg.git
[email protected]:mindspore/akg.git
mindspore
akg
akg
master

搜索帮助