()
| 79 | |
| 80 | |
| 81 | def test_move_to_device_end(): |
| 82 | test_data_shape = [1, 3, 0, 4] |
| 83 | |
| 84 | def get_data(): |
| 85 | out = [np.empty(test_data_shape, dtype=np.uint8) for _ in range(batch_size)] |
| 86 | return out |
| 87 | |
| 88 | pipe = Pipeline(batch_size=batch_size, num_threads=3, device_id=None) |
| 89 | outs = fn.external_source(source=get_data) |
| 90 | pipe.set_outputs(outs.gpu()) |
| 91 | assert_raises( |
| 92 | RuntimeError, |
| 93 | pipe.build, |
| 94 | glob="The pipeline requires a CUDA-capable GPU but *", |
| 95 | ) |
| 96 | |
| 97 | |
| 98 | def test_move_to_device_middle(): |
nothing calls this directly
no test coverage detected