(self, tmpdir, single_submit, overlap_events)
| 74 | set_dist_env = False |
| 75 | |
| 76 | def test_parallel_read(self, tmpdir, single_submit, overlap_events): |
| 77 | |
| 78 | h = GDSBuilder().load().gds_handle(BLOCK_SIZE, QUEUE_DEPTH, single_submit, overlap_events, IO_PARALLEL) |
| 79 | |
| 80 | gds_buffer = torch.empty(IO_SIZE, dtype=torch.uint8, device=get_accelerator().device_name()) |
| 81 | h.pin_device_tensor(gds_buffer) |
| 82 | |
| 83 | _validate_handle_state(h, single_submit, overlap_events) |
| 84 | |
| 85 | ref_file, _ = _do_ref_write(tmpdir) |
| 86 | read_status = h.sync_pread(gds_buffer, ref_file, 0) |
| 87 | assert read_status == 1 |
| 88 | |
| 89 | with open(ref_file, 'rb') as f: |
| 90 | ref_buffer = list(f.read()) |
| 91 | assert ref_buffer == gds_buffer.tolist() |
| 92 | |
| 93 | h.unpin_device_tensor(gds_buffer) |
| 94 | |
| 95 | def test_async_read(self, tmpdir, single_submit, overlap_events): |
| 96 |
nothing calls this directly
no test coverage detected