def do_set_libdevice(): os.environ['NUMBAPRO_LIBDEVICE'] = 'nbp_libdevice' return True, _get_libdevice_path_decision()
def do_clear_envs(): remove_env('CUDA_HOME') remove_env('NUMBAPRO_LIBDEVICE') remove_env('NUMBAPRO_CUDALIB') return True, _get_libdevice_path_decision()
def do_set_cuda_home(): os.environ['CUDA_HOME'] = os.path.join('mycudahome') _fake_non_conda_env() return True, _get_libdevice_path_decision()
def do_set_libdevice(): os.environ['NUMBAPRO_LIBDEVICE'] = 'nbp_libdevice' return True, _get_libdevice_path_decision()
def do_set_cuda_home(): os.environ['CUDA_HOME'] = os.path.join('mycudahome') _fake_non_conda_env() return True, _get_libdevice_path_decision()
def do_clear_envs(): remove_env('CUDA_HOME') remove_env('NUMBAPRO_LIBDEVICE') remove_env('NUMBAPRO_CUDALIB') return True, _get_libdevice_path_decision()
def do_set_libdevice(): os.environ["NUMBAPRO_LIBDEVICE"] = "nbp_libdevice" return True, _get_libdevice_path_decision()
def do_clear_envs(): remove_env('CUDA_HOME') remove_env('CUDA_PATH') return True, _get_libdevice_path_decision()