| 1234567891011121314151617181920212223 |
- # ------------------------------------------------------------------
- # Copyright (c) 2023 PyInstaller Development Team.
- #
- # This file is distributed under the terms of the GNU General Public
- # License (version 2.0 or later).
- #
- # The full license is available in LICENSE, distributed with
- # this software.
- #
- # SPDX-License-Identifier: GPL-2.0-or-later
- # ---------------------------------------------------
- from PyInstaller.utils.hooks import collect_dynamic_libs
- # bitsandbytes contains several extensions for CPU and different CUDA versions: libbitsandbytes_cpu.so,
- # libbitsandbytes_cuda110_nocublaslt.so, libbitsandbytes_cuda110.so, etc. At build-time, we could query the
- # `bitsandbytes.cextension.setup` and its `binary_name` attribute for the extension that is in use. However, if the
- # build system does not have CUDA available, this would automatically mean that we will not collect any of the CUDA
- # libs. So for now, we collect them all.
- binaries = collect_dynamic_libs("bitsandbytes")
- # bitsandbytes uses triton's JIT module, which requires access to source .py files.
- module_collection_mode = 'pyz+py'
|
