cubecl-macros 0.5.0

Procedural macros for CubeCL
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

use std::env;

// Allow overriding nightly macro features on the end user side without having to propagate the feature
// everywhere
fn main() {
    println!("cargo::rustc-check-cfg=cfg(debug_symbols)");
    println!("cargo::rustc-check-cfg=cfg(nightly)");
    println!("cargo:rerun-if-env-changed=CUBECL_DEBUG");
    println!("cargo:rerun-if-env-changed=CUBECL_DEBUG_NIGHTLY");

    let debug_feature_enabled = env::var("CARGO_FEATURE_DEBUG_SYMBOLS").is_ok();
    let debug_override_enabled = env::var("CUBECL_DEBUG").is_ok();
    let debug_enabled = debug_feature_enabled || debug_override_enabled;

    let nightly_feature_enabled = env::var("CARGO_FEATURE_NIGHTLY").is_ok();
    let nightly_override_enabled = env::var("CUBECL_DEBUG_NIGHTLY").is_ok();
    let nightly_enabled = nightly_feature_enabled || nightly_override_enabled;

    if debug_enabled || nightly_enabled {
        println!("cargo:rustc-cfg=debug_symbols");
    }

    if nightly_enabled {
        println!("cargo:rustc-cfg=nightly");
    }
}