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

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

use crate::common::event_attributes::GenericEventAttributes;
use crate::common::GenericMessage;

/// Generic event argument type that can be used in both global and domain contexts
#[derive(Debug, Clone)]
pub enum GenericEventArgument<M, A> {
    /// Holds a Message.
    Message(M),
    /// Holds an `EventAttributes`.
    Attributes(A),
}

impl<C, M> From<GenericEventAttributes<C, M>>
    for GenericEventArgument<M, GenericEventAttributes<C, M>>
{
    fn from(value: GenericEventAttributes<C, M>) -> Self {
        match value {
            GenericEventAttributes {
                category: None,
                color: None,
                message: Some(m),
                payload: None,
            } => GenericEventArgument::Message(m),
            attr => GenericEventArgument::Attributes(attr),
        }
    }
}

impl<C, R, T: Into<GenericMessage<R>>> From<T>
    for GenericEventArgument<GenericMessage<R>, GenericEventAttributes<C, GenericMessage<R>>>
{
    fn from(value: T) -> Self {
        GenericEventArgument::Message(value.into())
    }
}