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

//! Opaque handle wrappers for vendor SDK objects.
//!
//! Backends embed their raw cuBLAS / rocBLAS / MPS / QNN handles in these
//! transparent newtypes so that downstream code never accidentally crosses
//! backend boundaries with a raw pointer.

use core::marker::PhantomData;
use core::ptr::NonNull;

#[repr(transparent)]
pub struct Handle<Tag> {
    raw: NonNull<()>,
    _t: PhantomData<fn() -> Tag>,
}

impl<Tag> Handle<Tag> {
    /// # Safety
    /// `ptr` must be a valid handle of the right vendor type.
    #[inline(always)]
    pub const unsafe fn from_raw(ptr: NonNull<()>) -> Self {
        Self {
            raw: ptr,
            _t: PhantomData,
        }
    }

    #[inline(always)]
    pub const fn as_ptr(&self) -> NonNull<()> {
        self.raw
    }
}

unsafe impl<Tag: Send> Send for Handle<Tag> {}
unsafe impl<Tag: Sync> Sync for Handle<Tag> {}