def test_copy_from_gpu(): set_cuda_active(True) src = Variable(rand((100, ))) src.to_gpu() dest = Variable(rand((100, ))) dest.to_gpu() dest.copy_from(src) close(src, dest) close(src._gpu.new_array(), dest._gpu.new_array())
def test_copy_from_another_gpu(): set_cuda_active(True) src = Variable(rand((100, ))) src.to_gpu() with use_device(1): dest = Variable(rand((100, ))) dest.to_gpu() dest.copy_from(src) close(src, dest) close(src._gpu.new_array(), dest._gpu.new_array())