j2k-cuda 0.8.1

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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57

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

use j2k_core::HtGpuJobPassBucket;
use j2k_cuda_runtime::{
    CudaHtj2kCleanupTarget, CudaHtj2kDecodeResources, CudaQueuedHtj2kCleanup,
    CudaQueuedHtj2kCleanupGroup,
};

use super::super::super::super::{cuda_error, CudaBufferPool, CudaContext, Error};

#[expect(
    clippy::too_many_arguments,
    reason = "the CUDA enqueue boundary keeps resources, targets, pass route, pool, host budget, and status range explicit"
)]
pub(super) fn enqueue_chunk_kernel(
    context: &CudaContext,
    resources: &CudaHtj2kDecodeResources,
    targets: &[CudaHtj2kCleanupTarget<'_>],
    bucket: HtGpuJobPassBucket,
    pool: &CudaBufferPool,
    live_host_bytes: usize,
    status_group: &CudaQueuedHtj2kCleanupGroup,
    status_offset: usize,
) -> Result<CudaQueuedHtj2kCleanup, Error> {
    // SAFETY: the caller retains uploaded descriptors and every disjoint
    // coefficient target through status-group completion.
    let cleanup = unsafe {
        match bucket {
            HtGpuJobPassBucket::CleanupOnly => context
                .decode_htj2k_codeblocks_cleanup_dequantize_multi_enqueue_into_status_group(
                    resources,
                    targets,
                    pool,
                    live_host_bytes,
                    status_group,
                    status_offset,
                ),
            HtGpuJobPassBucket::SigProp | HtGpuJobPassBucket::MagRef => context
                .decode_htj2k_codeblocks_cleanup_multi_enqueue_into_status_group(
                    resources,
                    targets,
                    pool,
                    live_host_bytes,
                    status_group,
                    status_offset,
                ),
        }
    }
    .map_err(cuda_error)?;
    if bucket != HtGpuJobPassBucket::CleanupOnly {
        // SAFETY: cleanup retains its descriptors, and the same default
        // stream orders dequantization before later IDWT.
        unsafe { context.j2k_dequantize_queued_htj2k_cleanup_enqueue(&cleanup) }
            .map_err(cuda_error)?;
    }
    Ok(cleanup)
}