def run_test(): if not test(): return False nvvm = NVVM() print("NVVM version", nvvm.get_version()) return nvvm.get_version() is not None
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
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
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
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'))
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)
import sys from numba.cuda.cudadrv.libs import test sys.exit(0 if test() else 1)