ironaccelerator-core 2.0.0

Shared types, errors, and capability flags for IronAccelerator backends.
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

//! Vendor-neutral device descriptors.

use crate::{backend::BackendKind, capability::Capability};

#[cfg(not(feature = "std"))]
use alloc::string::String;
#[cfg(feature = "std")]
use std::string::String;

#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
pub struct DeviceId {
    pub backend: BackendKind,
    /// Backend-local ordinal (driver enumeration index).
    pub ordinal: u32,
}

#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
#[non_exhaustive]
pub enum Vendor {
    Nvidia,
    Amd,
    Apple,
    Qualcomm,
    Intel,
    Google,
    Aws,
    Other,
}

#[derive(Debug, Clone)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
pub struct DeviceDescriptor {
    pub id: DeviceId,
    pub vendor: Vendor,
    /// Marketing name (e.g. "NVIDIA H100 80GB HBM3").
    pub name: String,
    /// Vendor architecture string (e.g. "sm_90a", "gfx942", "Apple9", "v75").
    pub arch: String,
    pub total_memory_bytes: u64,
    pub multiprocessor_count: u32,
    pub clock_khz: u32,
    pub capability: Capability,
}

/// Live device handle. Concrete backends extend this trait with launch APIs.
pub trait Device: Send + Sync {
    fn descriptor(&self) -> &DeviceDescriptor;

    fn id(&self) -> DeviceId {
        self.descriptor().id
    }
}