代码拉取完成,页面将自动刷新
# 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',
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'
]
)
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。