File size: 1,718 Bytes
ff582c5 | 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 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 | # SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0
import sys
import platform
from setuptools import find_packages, setup
from wheel.bdist_wheel import bdist_wheel as _bdist_wheel
def _fetch_requirements(path):
with open(path, "r") as fd:
return [r.strip() for r in fd.readlines() if r.strip() and not r.startswith("#")]
# Custom wheel class to modify the wheel name
class bdist_wheel(_bdist_wheel):
def finalize_options(self):
_bdist_wheel.finalize_options(self)
self.root_is_pure = False
def get_tag(self):
python_version = f"cp{sys.version_info.major}{sys.version_info.minor}"
abi_tag = f"{python_version}"
if platform.system() == "Linux":
platform_tag = "manylinux1_x86_64"
else:
platform_tag = platform.system().lower()
return python_version, abi_tag, platform_tag
# Setup configuration
setup(
author="slime Team",
name="slime",
version="0.1.0",
packages=find_packages(include=["slime*", "slime_plugins*"]),
include_package_data=True,
install_requires=_fetch_requirements("requirements.txt"),
extras_require={
"fsdp": [
"torch>=2.0",
]
},
python_requires=">=3.10",
classifiers=[
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.11",
"Programming Language :: Python :: 3.12",
"Environment :: GPU :: NVIDIA CUDA",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
"Topic :: System :: Distributed Computing",
],
cmdclass={"bdist_wheel": bdist_wheel},
)
|