def do_set_nvvm(): os.environ['NUMBAPRO_NVVM'] = 'nbp_nvvm' return True, _get_nvvm_path_decision()
def do_set_cuda_home(): os.environ['CUDA_HOME'] = os.path.join('mycudahome') _fake_non_conda_env() return True, _get_nvvm_path_decision()
def do_set_cuda_lib(): os.environ['NUMBAPRO_CUDALIB'] = 'nbp_cudalib' return True, _get_nvvm_path_decision()
def do_set_nvvm(): os.environ['NUMBAPRO_NVVM'] = 'nbp_nvvm' return True, _get_nvvm_path_decision()
def do_clear_envs(): remove_env('CUDA_HOME') remove_env('NUMBAPRO_CUDALIB') remove_env('NUMBAPRO_NVVM') return True, _get_nvvm_path_decision()
def do_set_cuda_home(): os.environ['CUDA_HOME'] = os.path.join('mycudahome') _fake_non_conda_env() return True, _get_nvvm_path_decision()
def do_set_cuda_lib(): os.environ['NUMBAPRO_CUDALIB'] = 'nbp_cudalib' return True, _get_nvvm_path_decision()
def do_clear_envs(): remove_env('CUDA_HOME') remove_env('NUMBAPRO_CUDALIB') remove_env('NUMBAPRO_NVVM') return True, _get_nvvm_path_decision()
def do_set_nvvm(): os.environ["NUMBAPRO_NVVM"] = "nbp_nvvm" return True, _get_nvvm_path_decision()
def do_set_cuda_lib(): os.environ["NUMBAPRO_CUDALIB"] = "nbp_cudalib" return True, _get_nvvm_path_decision()
def do_clear_envs(): remove_env("CUDA_HOME") remove_env("NUMBAPRO_CUDALIB") remove_env("NUMBAPRO_NVVM") return True, _get_nvvm_path_decision()
def do_clear_envs(): remove_env('CUDA_HOME') remove_env('CUDA_PATH') return True, _get_nvvm_path_decision()