cubecl-cpu 0.11.0-pre.1

CPU runtime for CubeCL
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

#[cfg(any(target_os = "android", target_os = "linux"))]
mod linux;
#[cfg(any(target_os = "android", target_os = "linux"))]
pub use linux::get_active_cores;
#[cfg(any(target_os = "android", target_os = "linux"))]
pub use linux::set_for_current;

#[cfg(target_os = "windows")]
mod windows;
#[cfg(target_os = "windows")]
pub use windows::get_active_cores;
#[cfg(target_os = "windows")]
pub use windows::set_for_current;

#[repr(transparent)]
#[derive(Copy, Clone, Debug, PartialEq, Eq, PartialOrd, Ord, Hash)]
pub struct CoreId(usize);

#[cfg(not(any(target_os = "linux", target_os = "android", target_os = "windows",)))]
#[inline]
pub fn get_active_cores() -> impl Iterator<Item = CoreId> {
    let cores = std::thread::available_parallelism().map_or(1, |n| n.get());
    (0..cores).map(CoreId)
}

#[cfg(not(any(target_os = "linux", target_os = "android", target_os = "windows",)))]
#[inline]
pub fn set_for_current(_core_id: CoreId) {}