bitsandbytes/setup.py at fix-replace-linear-post-processing · ErenAta16/bitsandbytes · GitHub

History

49 lines (39 loc) · 1.52 KB

Raw

# Copyright (c) Facebook, Inc. and its affiliates.

# This source code is licensed under the MIT license found in the

# LICENSE file in the root directory of this source tree.

from distutils.errors import DistutilsModuleError

import os

from warnings import warn

from setuptools import find_packages, setup

from setuptools.command.build_py import build_py

from setuptools.dist import Distribution

# Tested with wheel v0.29.0

class BinaryDistribution(Distribution):

def has_ext_modules(self):

return True

class ExtBuildPy(build_py):

def run(self):

if os.environ.get("BNB_SKIP_CMAKE", "").lower() in ("1", "true", "yes"):

print("skipping CMake build")

else:

# build_cmake needs to be called prior to build_py, as the latter

# collects the files output into the package directory.

try:

self.run_command("build_cmake")

except DistutilsModuleError:

warn(

"scikit-build-core not installed, CMake will not be invoked automatically. "

"Please install scikit-build-core or run CMake manually to build extensions."

)

super().run()

cmdclass = {"build_py": ExtBuildPy}

setup_kwargs = {

"version": "0.50.1.dev0",

"packages": find_packages(),

"distclass": BinaryDistribution,

"cmdclass": {"build_py": ExtBuildPy},

}

if os.environ.get("BNB_SKIP_CMAKE", "").lower() not in ("1", "true", "yes"):

setup_kwargs["cmake_source_dir"] = "."

setup(**setup_kwargs)

bitsandbytes/setup.py at fix-replace-linear-post-processing · ErenAta16/bitsandbytes · GitHub

Expand file tree

Breadcrumbs

setup.py

Latest commit

History

Breadcrumbs

setup.py

File metadata and controls