Esempio n. 1
0
def _get_nvvm_path_decision():
    options = [
        ('NUMBAPRO_NVVM', get_numbapro_envvar('NUMBAPRO_NVVM')),
        ('NUMBAPRO_CUDALIB', get_numbapro_envvar('NUMBAPRO_CUDALIB')),
        ('Conda environment', get_conda_ctk()),
        ('CUDA_HOME', get_cuda_home(*_nvvm_lib_dir())),
        ('System', get_system_ctk(*_nvvm_lib_dir())),
    ]
    by, path = _find_valid_path(options)
    return by, path
Esempio n. 2
0
def _get_libdevice_path_decision():
    options = [
        ('NUMBAPRO_LIBDEVICE', get_numbapro_envvar('NUMBAPRO_LIBDEVICE')),
        ('NUMBAPRO_CUDALIB', get_numbapro_envvar('NUMBAPRO_CUDALIB')),
        ('Conda environment', get_conda_ctk()),
        ('CUDA_HOME', get_cuda_home('nvvm', 'libdevice')),
        ('System', get_system_ctk('nvvm', 'libdevice')),
    ]
    by, libdir = _find_valid_path(options)
    return by, libdir
Esempio n. 3
0
def _get_nvvm_path_decision():
    options = [
        ("NUMBAPRO_NVVM", get_numbapro_envvar("NUMBAPRO_NVVM")),
        ("NUMBAPRO_CUDALIB", get_numbapro_envvar("NUMBAPRO_CUDALIB")),
        ("Conda environment", get_conda_ctk()),
        ("CUDA_HOME", get_cuda_home(*_nvvm_lib_dir())),
        ("System", get_system_ctk(*_nvvm_lib_dir())),
    ]
    by, path = _find_valid_path(options)
    return by, path
Esempio n. 4
0
def _get_nvvm_path_decision():
    options = [
        ('NUMBAPRO_NVVM', get_numbapro_envvar('NUMBAPRO_NVVM')),
        ('NUMBAPRO_CUDALIB', get_numbapro_envvar('NUMBAPRO_CUDALIB')),
        ('Conda environment', get_conda_ctk()),
        ('CUDA_HOME', get_cuda_home(*_nvvm_lib_dir())),
        ('System', get_system_ctk(*_nvvm_lib_dir())),
    ]
    by, libdir = _find_valid_path(options)
    return by, libdir
Esempio n. 5
0
def _get_libdevice_path_decision():
    options = [
        ('NUMBAPRO_LIBDEVICE', get_numbapro_envvar('NUMBAPRO_LIBDEVICE')),
        ('NUMBAPRO_CUDALIB', get_numbapro_envvar('NUMBAPRO_CUDALIB')),
        ('Conda environment', get_conda_ctk()),
        ('CUDA_HOME', get_cuda_home('nvvm', 'libdevice')),
        ('System', get_system_ctk('nvvm', 'libdevice')),
    ]
    by, libdir = _find_valid_path(options)
    return by, libdir
Esempio n. 6
0
def _get_libdevice_path_decision():
    options = [
        ("NUMBAPRO_LIBDEVICE", get_numbapro_envvar("NUMBAPRO_LIBDEVICE")),
        ("NUMBAPRO_CUDALIB", get_numbapro_envvar("NUMBAPRO_CUDALIB")),
        ("Conda environment", get_conda_ctk()),
        ("CUDA_HOME", get_cuda_home("nvvm", "libdevice")),
        ("System", get_system_ctk("nvvm", "libdevice")),
        ("Debian package", get_debian_pkg_libdevice()),
    ]
    by, libdir = _find_valid_path(options)
    return by, libdir