j2k-cuda 0.8.0

CUDA adapter for resident HTJ2K decode/encode and shared JPEG 2000 stages
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31

// SPDX-License-Identifier: MIT OR Apache-2.0

use j2k::PreparedBatchGroup;
use j2k_cuda::{CudaBatchDecoder, SubmittedCudaExternalBatch};
use j2k_cuda_runtime::{CudaContext, CudaDeviceBuffer, CudaExternalDeviceBufferViewMut};

pub(super) unsafe fn submit_external_for_test(
    decoder: &mut CudaBatchDecoder,
    group: &PreparedBatchGroup,
    context: &CudaContext,
    allocation: &mut CudaDeviceBuffer,
) -> SubmittedCudaExternalBatch {
    let ptr = allocation.device_ptr();
    let len = allocation.byte_len();
    // SAFETY: the caller guarantees `allocation` stays live and inaccessible
    // until the returned completion owner is retired.
    let mut destination = unsafe {
        CudaExternalDeviceBufferViewMut::from_raw_parts(
            context,
            ptr,
            len,
            std::mem::align_of::<u8>(),
            allocation,
        )
    }
    .expect("external destination view");
    // SAFETY: forwarded from this helper's caller; the view exclusively
    // borrows the live allocation for submission.
    unsafe { decoder.submit_batch_into(group, &mut destination) }
        .expect("submit external test batch")
}