test_ndarray_get.py 文件源码

def test_get_multigpu(self, dtype):
        with cuda.Device(1):
            src = testing.shaped_arange((2, 3), xp=cupy, dtype=dtype)
            src = cupy.asfortranarray(src)
        with cuda.Device(0):
            dst = src.get()
        expected = testing.shaped_arange((2, 3), xp=numpy, dtype=dtype)
        np_testing.assert_array_equal(dst, expected)