hodu_cuda_kernels 0.2.3

hodu cuda kernels
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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113

//! std/no_std compatibility layer
//!
//! Internal module for handling differences between std and no-std environments,
//! and cudarc API compatibility.

#![allow(unused_imports)]

// Basic types and formatting
#[cfg(not(feature = "std"))]
pub use alloc::{
    boxed::Box,
    format,
    string::{String, ToString},
    vec,
    vec::Vec,
};

#[cfg(feature = "std")]
pub use std::{
    boxed::Box,
    format,
    string::{String, ToString},
    vec,
    vec::Vec,
};

// Collections
#[cfg(not(feature = "std"))]
pub use alloc::collections::{BTreeMap as HashMap, BTreeSet as HashSet};

#[cfg(feature = "std")]
pub use std::collections::{HashMap, HashSet};

// Synchronization primitives
#[cfg(not(feature = "std"))]
pub use alloc::sync::{Arc, Weak};

#[cfg(feature = "std")]
pub use std::sync::{Arc, Weak};

#[cfg(not(feature = "std"))]
pub use spin::{Mutex, RwLock};

#[cfg(feature = "std")]
pub use std::sync::{Mutex, RwLock};

// Core traits and functions
#[cfg(not(feature = "std"))]
pub use core::{
    cell::{Cell, RefCell},
    fmt,
    hash::{Hash, Hasher},
    ops,
};

#[cfg(feature = "std")]
pub use std::{
    cell::{Cell, RefCell},
    fmt,
    hash::{Hash, Hasher},
    ops,
};

// Lazy/OnceLock compatibility
#[cfg(not(feature = "std"))]
pub use spin::Lazy;

#[cfg(feature = "std")]
pub use std::sync::LazyLock as Lazy;

// RwLock compatibility wrapper
pub struct RwLockGuard<T>(T);

impl<T> core::ops::Deref for RwLockGuard<T> {
    type Target = T;
    fn deref(&self) -> &Self::Target {
        &self.0
    }
}

impl<T> core::ops::DerefMut for RwLockGuard<T> {
    fn deref_mut(&mut self) -> &mut Self::Target {
        &mut self.0
    }
}

// Extension trait for RwLock to provide Result-returning API
pub trait RwLockExt<T> {
    fn read_compat(&self) -> Result<RwLockGuard<impl core::ops::Deref<Target = T> + '_>, String>;
    fn write_compat(&self) -> Result<RwLockGuard<impl core::ops::DerefMut<Target = T> + '_>, String>;
}

#[cfg(feature = "std")]
impl<T> RwLockExt<T> for RwLock<T> {
    fn read_compat(&self) -> Result<RwLockGuard<impl core::ops::Deref<Target = T> + '_>, String> {
        self.read().map(RwLockGuard).map_err(|e| format!("{:?}", e))
    }

    fn write_compat(&self) -> Result<RwLockGuard<impl core::ops::DerefMut<Target = T> + '_>, String> {
        self.write().map(RwLockGuard).map_err(|e| format!("{:?}", e))
    }
}

#[cfg(not(feature = "std"))]
impl<T> RwLockExt<T> for RwLock<T> {
    fn read_compat(&self) -> Result<RwLockGuard<impl core::ops::Deref<Target = T> + '_>, String> {
        Ok(RwLockGuard(self.read()))
    }

    fn write_compat(&self) -> Result<RwLockGuard<impl core::ops::DerefMut<Target = T> + '_>, String> {
        Ok(RwLockGuard(self.write()))
    }
}