33 lines
1.0 KiB
Rust
33 lines
1.0 KiB
Rust
extern crate rustc_version;
|
|
use rustc_version::{version_meta, Channel};
|
|
|
|
fn main() {
|
|
#[cfg(any(target_arch = "x86", target_arch = "x86_64"))]
|
|
{
|
|
if is_x86_feature_detected!("avx") {
|
|
println!("cargo:rustc-cfg=build_target_feature_avx");
|
|
}
|
|
if is_x86_feature_detected!("avx2") {
|
|
println!("cargo:rustc-cfg=build_target_feature_avx2");
|
|
}
|
|
}
|
|
|
|
// Copied and adapted from
|
|
// https://github.com/Kimundi/rustc-version-rs/blob/1d692a965f4e48a8cb72e82cda953107c0d22f47/README.md#example
|
|
// Licensed under Apache-2.0 + MIT
|
|
match version_meta().unwrap().channel {
|
|
Channel::Stable => {
|
|
println!("cargo:rustc-cfg=RUSTC_WITHOUT_SPECIALIZATION");
|
|
}
|
|
Channel::Beta => {
|
|
println!("cargo:rustc-cfg=RUSTC_WITHOUT_SPECIALIZATION");
|
|
}
|
|
Channel::Nightly => {
|
|
println!("cargo:rustc-cfg=RUSTC_WITH_SPECIALIZATION");
|
|
}
|
|
Channel::Dev => {
|
|
println!("cargo:rustc-cfg=RUSTC_WITH_SPECIALIZATION");
|
|
}
|
|
}
|
|
}
|