def unuse():
"""
This undo what was done by the call to
use('gpu[0-9]', default_to_move_computation_to_gpu=True,
move_shared_float32_to_gpu=True,
enable_cuda=True)
This is used in Pylearn2 tests to enable/disable the GPU when needed.
After this call, the rest of Theano think the GPU shouldn't be used by default.
"""
global cuda_enabled
cuda_enabled = False
handle_shared_float32(False)
optdb.remove_tags('gpu_opt', 'fast_run', 'inplace')
optdb.remove_tags('gpu_after_fusion', 'fast_run', 'inplace')
def unuse():
"""
This undo what was done by the call to
use('gpu[0-9]', default_to_move_computation_to_gpu=True,
move_shared_float32_to_gpu=True,
enable_cuda=True)
This is used in Pylearn2 tests to enable/disable the GPU when needed.
After this call, the rest of Theano think the GPU shouldn't be used by default.
"""
global cuda_enabled
cuda_enabled = False
handle_shared_float32(False)
optdb.remove_tags("gpu_opt", "fast_run", "inplace")
optdb.remove_tags("gpu_after_fusion", "fast_run", "inplace")
