()
| 456 | |
| 457 | |
| 458 | def test_pinned_input_hw_decoder(): |
| 459 | file_root = os.path.join(test_data_root, good_path, "jpeg") |
| 460 | |
| 461 | @pipeline_def(batch_size=128, device_id=0, num_threads=4) |
| 462 | def pipe(): |
| 463 | encoded, _ = fn.readers.file(file_root=file_root) |
| 464 | encoded_gpu = encoded.gpu() |
| 465 | # encoded.gpu() should make encoded pinned |
| 466 | decoded = fn.decoders.image(encoded, device="mixed") |
| 467 | return decoded, encoded_gpu |
| 468 | |
| 469 | p = pipe() |
| 470 | p.run() |
| 471 | |
| 472 | |
| 473 | def test_tiff_palette(): |