| 24 | |
| 25 | |
| 26 | class ExternalSourcePipe(Pipeline): |
| 27 | def __init__(self, batch_size, data, use_copy_kernel=False): |
| 28 | super(ExternalSourcePipe, self).__init__(batch_size, 1, 0) |
| 29 | self.output = ops.ExternalSource(device="gpu") |
| 30 | self.data = data |
| 31 | self.use_copy_kernel = use_copy_kernel |
| 32 | |
| 33 | def define_graph(self): |
| 34 | self.out = self.output() |
| 35 | return self.out |
| 36 | |
| 37 | def iter_setup(self): |
| 38 | self.feed_input(self.out, self.data, use_copy_kernel=self.use_copy_kernel) |
| 39 | |
| 40 | |
| 41 | def test_tensorlist_getitem_gpu(): |
no outgoing calls