Beispiel #1
0
 def do_set_libdevice():
     os.environ['NUMBAPRO_LIBDEVICE'] = 'nbp_libdevice'
     return True, _get_libdevice_path_decision()
Beispiel #2
0
 def do_clear_envs():
     remove_env('CUDA_HOME')
     remove_env('NUMBAPRO_LIBDEVICE')
     remove_env('NUMBAPRO_CUDALIB')
     return True, _get_libdevice_path_decision()
Beispiel #3
0
 def do_set_cuda_home():
     os.environ['CUDA_HOME'] = os.path.join('mycudahome')
     _fake_non_conda_env()
     return True, _get_libdevice_path_decision()
Beispiel #4
0
 def do_set_libdevice():
     os.environ['NUMBAPRO_LIBDEVICE'] = 'nbp_libdevice'
     return True, _get_libdevice_path_decision()
Beispiel #5
0
 def do_set_cuda_home():
     os.environ['CUDA_HOME'] = os.path.join('mycudahome')
     _fake_non_conda_env()
     return True, _get_libdevice_path_decision()
Beispiel #6
0
 def do_clear_envs():
     remove_env('CUDA_HOME')
     remove_env('NUMBAPRO_LIBDEVICE')
     remove_env('NUMBAPRO_CUDALIB')
     return True, _get_libdevice_path_decision()
Beispiel #7
0
 def do_set_libdevice():
     os.environ["NUMBAPRO_LIBDEVICE"] = "nbp_libdevice"
     return True, _get_libdevice_path_decision()
Beispiel #8
0
 def do_clear_envs():
     remove_env('CUDA_HOME')
     remove_env('CUDA_PATH')
     return True, _get_libdevice_path_decision()