nvtx 2.0.0

NVIDIA Tools Extension SDK
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
58
59
60
61
62

// SPDX-FileCopyrightText: Copyright (c) 2024-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

use crate::TypeValueEncodable;

mod generic;
pub use generic::GenericIdentifier;

#[cfg(target_family = "unix")]
mod pthread;
#[cfg(target_family = "unix")]
pub use pthread::PThreadIdentifier;

#[cfg(feature = "cuda_runtime")]
mod cuda_runtime;
#[cfg(feature = "cuda_runtime")]
pub use cuda_runtime::CudaRuntimeIdentifier;

#[cfg(feature = "cuda")]
mod cuda;
#[cfg(feature = "cuda")]
pub use cuda::CudaIdentifier;

/// Identifier used for supported resource types
#[non_exhaustive]
pub enum Identifier {
    /// Generic identifier
    Generic(GenericIdentifier),
    /// PThread-specific identifier
    #[cfg(target_family = "unix")]
    PThread(PThreadIdentifier),
    /// CUDA specific identifier
    #[cfg(feature = "cuda")]
    Cuda(CudaIdentifier),
    /// CUDA runtime specific identifier
    #[cfg(feature = "cuda_runtime")]
    CudaRuntime(CudaRuntimeIdentifier),
}

impl TypeValueEncodable for Identifier {
    type Type = u32;
    type Value = nvtx_sys::ResourceAttributesIdentifier;

    fn encode(&self) -> (Self::Type, Self::Value) {
        match self {
            Identifier::Generic(g) => g.encode(),
            #[cfg(target_family = "unix")]
            Identifier::PThread(p) => p.encode(),
            #[cfg(feature = "cuda")]
            Identifier::Cuda(c) => c.encode(),
            #[cfg(feature = "cuda_runtime")]
            Identifier::CudaRuntime(c) => c.encode(),
        }
    }

    fn default_encoding() -> (Self::Type, Self::Value) {
        (
            nvtx_sys::resource_type::UNKNOWN,
            Self::Value { ullValue: 0 },
        )
    }
}