61 lines
1.7 KiB
Python
61 lines
1.7 KiB
Python
# -*- coding: utf-8 -*-
|
||
import os
|
||
from pathlib import Path
|
||
from setuptools import setup, find_packages
|
||
|
||
# 基础目录
|
||
BASE = Path(__file__).parent
|
||
SRC_DIR = BASE / "src"
|
||
|
||
# 收集 netrans 包（不包含 acuitylib）
|
||
netrans_packages = find_packages(
|
||
where=str(SRC_DIR),
|
||
include=["netrans", "netrans.*"]
|
||
)
|
||
|
||
setup(
|
||
name="netrans",
|
||
version="6.33.6",
|
||
description="Netrans inference SDK - Neural Network Transformation Toolkit for PNNA Chips",
|
||
author="Your Name",
|
||
author_email="you@example.com",
|
||
url="https://github.com/yourname/netrans",
|
||
license="Proprietary",
|
||
python_requires=">=3.10",
|
||
|
||
# 关键：告诉 setuptools 去哪里找包
|
||
package_dir={
|
||
"": "src", # 所有包都在 src 下
|
||
},
|
||
packages=netrans_packages,
|
||
include_package_data=True,
|
||
zip_safe=False,
|
||
|
||
# 入口点 - 创建命令行脚本
|
||
entry_points={
|
||
'console_scripts': [
|
||
'netrans=netrans.cli:main',
|
||
'netrans-load=netrans.cli:main',
|
||
'netrans-quantize=netrans.cli:main',
|
||
'netrans-export=netrans.cli:main',
|
||
'netrans-dump=netrans.cli:main',
|
||
'netrans-inference=netrans.cli:main',
|
||
'netrans-quantize-hybrid=netrans.cli:main',
|
||
'netrans-add-prepost=netrans.cli:main',
|
||
'netrans-check-opset=netrans.cli:main',
|
||
'netrans-measure=netrans.cli:main',
|
||
],
|
||
},
|
||
|
||
# 依赖说明（acuity 需要单独安装）
|
||
extras_require={
|
||
'acuity': ['acuity>=6.33.0'],
|
||
},
|
||
|
||
classifiers=[
|
||
"Development Status :: 4 - Beta",
|
||
"Intended Audience :: Developers",
|
||
"Programming Language :: Python :: 3.10",
|
||
],
|
||
)
|