(_img_type, data_path, batch, device, threads, memory_stats=False)
| 72 | |
| 73 | |
| 74 | def run_decode(_img_type, data_path, batch, device, threads, memory_stats=False): |
| 75 | pipe = decoder_pipe( |
| 76 | data_path=data_path, |
| 77 | batch_size=batch, |
| 78 | num_threads=threads, |
| 79 | device_id=0, |
| 80 | device=device, |
| 81 | memory_stats=memory_stats, |
| 82 | prefetch_queue_depth=1, |
| 83 | ) |
| 84 | iters = math.ceil(pipe.epoch_size("Reader") / batch) |
| 85 | for _ in range(iters): |
| 86 | outs = pipe.run() |
| 87 | del outs |
| 88 | del pipe |
| 89 | |
| 90 | |
| 91 | def test_image_decoder(): |
nothing calls this directly
no test coverage detected