| 123456789101112131415161718192021222324252627282930 |
- # ------------------------------------------------------------------
- # 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_data_files
- from _pyinstaller_hooks_contrib.utils.nvidia_cuda import (
- collect_nvidia_cuda_binaries,
- create_symlink_suppression_patterns,
- )
- # Ensures that versioned .so files are collected
- binaries = collect_nvidia_cuda_binaries(__file__)
- # Prevent binary dependency analysis from creating symlinks to top-level application directory for shared libraries
- # from this package. Requires PyInstaller >= 6.11.0; no-op in earlier versions.
- bindepend_symlink_suppression = create_symlink_suppression_patterns(__file__)
- # Collect additional resources:
- # - ptxas executable (which strictly speaking, should be collected as a binary)
- # - nvvm/libdevice/libdevice.10.bc file
- # - C headers; assuming ptxas requires them - if that is not the case, we could filter them out.
- datas = collect_data_files('nvidia.cuda_nvcc')
|
