cubecl 0.9.0-pre.6

Multi-platform high-performance compute language extension for Rust.
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

use std::collections::BTreeMap;

fn main() {
    let enable_runtime = cfg!(feature = "test-runtime");

    println!("cargo:rustc-check-cfg=cfg(test_runtime_default)");
    println!("cargo:rustc-check-cfg=cfg(test_runtime_cpu)");
    println!("cargo:rustc-check-cfg=cfg(test_runtime_cuda)");
    println!("cargo:rustc-check-cfg=cfg(test_runtime_hip)");
    println!("cargo:rustc-check-cfg=cfg(test_runtime_wgpu)");

    if enable_runtime {
        let map = BTreeMap::from([
            ("cpu", cfg!(feature = "cpu")),
            ("cuda", cfg!(feature = "cuda")),
            ("hip", cfg!(feature = "hip")),
            ("wgpu", cfg!(feature = "wgpu")),
        ]);

        let enabled_features = map
            .iter()
            .filter(|(_, enabled)| **enabled)
            .map(|(k, _)| *k)
            .collect::<Vec<_>>();

        if enabled_features.is_empty() || enabled_features.len() > 1 {
            println!("cargo:rustc-cfg=test_runtime_default");
        } else {
            println!("cargo:rustc-cfg=test_runtime_{}", enabled_features[0]);
        }
    }
}