colossalai

1
from setuptools import find_packages, setup
2

3

4
def fetch_requirements(path):
5
    with open(path, "r") as fd:
6
        return [r.strip() for r in fd.readlines()]
7

8

9
def fetch_readme():
10
    with open("README.md", encoding="utf-8") as f:
11
        return f.read()
12

13

14
def fetch_version():
15
    with open("version.txt", "r") as f:
16
        return f.read().strip()
17

18

19
setup(
20
    name="colossal_moe",
21
    version=fetch_version(),
22
    packages=find_packages(
23
        exclude=(
24
            "tests",
25
            "benchmarks",
26
            "*.egg-info",
27
        )
28
    ),
29
    description="Colossal-AI MoE",
30
    long_description=fetch_readme(),
31
    long_description_content_type="text/markdown",
32
    license="Apache Software License 2.0",
33
    url="https://github.com/hpcaitech",
34
    install_requires=fetch_requirements("requirements.txt"),
35
    python_requires=">=3.6",
36
    classifiers=[
37
        "Programming Language :: Python :: 3",
38
        "License :: OSI Approved :: Apache Software License",
39
        "Environment :: GPU :: NVIDIA CUDA",
40
        "Topic :: Scientific/Engineering :: Artificial Intelligence",
41
        "Topic :: System :: Distributed Computing",
42
    ],
43
)
44