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())
