cubecl-metal 0.11.0

Metal 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
29
30
31
32
33
34
35
36
37
38
39
40

// Metal only exists on Apple platforms; elsewhere the crate compiles to
// nothing so workspace-wide builds and clippy work on any host. The Metal
// bindings it needs are target-gated in Cargo.toml to match.
#![cfg(target_vendor = "apple")]

pub mod compute;
pub mod device;
pub mod memory;
pub mod runtime;

pub use device::{MetalDevice, register_device};
pub use runtime::MetalRuntime;

pub(crate) type MetalCompiler = cubecl_cpp::shared::CppCompiler<cubecl_cpp::target::Metal>;

#[cfg(test)]
mod tests_bf16_cast;
#[cfg(test)]
mod tests_expm1;
#[cfg(test)]
mod tests_launch_errors;
#[cfg(test)]
mod tests_multistream;

#[cfg(test)]
mod tests {
    pub type TestRuntime = crate::MetalRuntime;

    use half::{bf16, f16};

    cubecl_core::testgen_all!(f32: [f16, bf16, f32], i32: [i8, i16, i32, i64], u32: [u8, u16, u32, u64]);
    cubecl_std::testgen!();
    cubecl_std::testgen_tensor_identity!([f16, f32, u32]);
    cubecl_std::testgen_quantized_view!(f32);
    cubecl_core::testgen_profiling!();
    // No `testgen_profiling_nested!`: a stream collects the command buffers of
    // one window at a time (`Stream::profiling`), so an inner `start_profile`
    // replaces the outer's collector and the outer window ends up measuring
    // nothing. Add it once the collector is a stack.
}