Python NVVM.get_version Exemples

Exemple #1

Fichier : run_test.py Projet : xmnlab/cudatoolkit-dev-feedstock

def run_test():

    if not test():
        return False
    nvvm = NVVM()
    print("NVVM version", nvvm.get_version())
    return nvvm.get_version() is not None

Exemple #2

def run_test():
    # on windows only nvvm is available to numba
    if sys.platform.startswith('win'):
        nvvm = NVVM()
        print("NVVM version", nvvm.get_version())
        return nvvm.get_version() is not None
    if not test():
        return False
    nvvm = NVVM()
    print("NVVM version", nvvm.get_version())
    # check pkg version matches lib pulled in
    gotlib = get_cudalib('cublas')
    lookfor = os.environ['PKG_VERSION']
    return lookfor in gotlib

Exemple #3

Fichier : run_test.py Projet : scopatz/cudatoolkit-feedstock

def run_test():
    # on windows only nvvm is available to numba
    if sys.platform.startswith('win'):
        nvvm = NVVM()
        print("NVVM version", nvvm.get_version())
        return nvvm.get_version() is not None
    if not test():
        return False
    nvvm = NVVM()
    print("NVVM version", nvvm.get_version())
    # check pkg version matches lib pulled in
    gotlib = get_cudalib('cublas')
    # check cufft b/c cublas has an incorrect version in 10.1 update 1
    gotlib = get_cudalib('cufft')
    return bool(get_cudalib('cublas')) and bool(get_cudalib('cufft'))

Exemple #4

Fichier : test_nvvm_driver.py Projet : ymarfoq/outilACVDesagregation

    def get_ptx(self):
        nvvm = NVVM()
        print(nvvm.get_version())

        if is64bit:
            return gpu64
        else:
            return gpu32

Exemple #5

Fichier : test_nvvm_driver.py Projet : winstonewert/numba

    def get_ptx(self):
        nvvm = NVVM()
        print(nvvm.get_version())

        if is64bit:
            return gpu64
        else:
            return gpu32

Exemple #6

def run_test():
    # on windows only nvvm is available to numba
    if sys.platform.startswith("win"):
        nvvm = NVVM()
        print("NVVM version", nvvm.get_version())
        return nvvm.get_version() is not None
    if not test():
        return False
    nvvm = NVVM()
    print("NVVM version", nvvm.get_version())

    extra_lib_tests = (
        "cublas",  # check pkg version matches lib pulled in
        "cufft",  # check cufft b/c cublas has an incorrect version in 10.1 update 1
        "cupti",  # check this is getting included
    )
    found_paths = [get_cudalib(lib) for lib in extra_lib_tests]
    print(*zip(extra_lib_tests, found_paths), sep="\n")

    return all(extra_lib_tests)

Exemple #7

Fichier : run_test.py Projet : Quansight/conda-recipe-cudatoolkit-dev

def run_test():
    if not test():
        return False
    nvvm = NVVM()
    print("NVVM version", nvvm.get_version())
    # check pkg version matches lib pulled in
    gotlib = get_cudalib('cublas')
    lookfor = '9.2'
    if sys.platform.startswith('win'):
        # windows libs have no dot
        lookfor = lookfor.replace('.', '')
    return lookfor in gotlib

Exemple #8

Fichier : run_test.py Projet : sklam/conda-recipe-cudatoolkit

def run_test():
    if not test():
        return False
    nvvm = NVVM()
    print("NVVM version", nvvm.get_version())
    # check pkg version matches lib pulled in
    gotlib = get_cudalib('cublas')
    lookfor = os.environ['PKG_VERSION']
    if sys.platform.startswith('win'):
        # windows libs have no dot
        lookfor = lookfor.replace('.', '')
    return lookfor in gotlib