()
| 343 | |
| 344 | |
| 345 | def test_tensor_gpu_from_cupy(): |
| 346 | def create_tmp(idx): |
| 347 | a = np.full((4, 4), idx) |
| 348 | a_gpu = cp.array(a, dtype=a.dtype) |
| 349 | return tensors.TensorGPU(a_gpu, "") |
| 350 | |
| 351 | out = [create_tmp(i) for i in range(4)] |
| 352 | for i, t in enumerate(out): |
| 353 | np.testing.assert_array_equal(np.array(t.as_cpu()), np.full((4, 4), i)) |
| 354 | |
| 355 | |
| 356 | def test_tensor_list_gpu_from_cupy(): |
nothing calls this directly
no test coverage detected