99 lines
3.1 KiB
Rust
99 lines
3.1 KiB
Rust
|
// Copyright 2015 The Rust Project Developers. See the COPYRIGHT
|
||
|
|
// file at the top-level directory of this distribution and at
|
||
|
|
// http://rust-lang.org/COPYRIGHT.
|
||
|
|
//
|
||
|
|
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
|
||
|
|
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
|
||
|
|
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
|
||
|
|
// option. This file may not be copied, modified, or distributed
|
||
|
|
// except according to those terms.
|
||
|
|
|
||
|
|
use syntax::{ast, attr};
|
||
|
|
use rustc::session::Session;
|
||
|
|
use syntax::parse::token::InternedString;
|
||
|
|
use syntax::parse::token::intern_and_get_ident as intern;
|
||
|
|
|
||
|
|
/// Add `target_feature = "..."` cfgs for a variety of platform
|
||
|
|
/// specific features (SSE, NEON etc.).
|
||
|
|
///
|
||
|
|
/// This uses a scheme similar to that employed by clang: reimplement
|
||
|
|
/// the target feature knowledge. *Theoretically* we could query LLVM
|
||
|
|
/// since that has perfect knowledge about what things are enabled in
|
||
|
|
/// code-generation, however, it is extremely non-obvious how to do
|
||
|
|
/// this successfully. Each platform defines a subclass of a
|
||
|
|
/// SubtargetInfo, which knows all this information, but the ways to
|
||
|
|
/// query them do not seem to be public.
|
||
|
|
pub fn add_configuration(cfg: &mut ast::CrateConfig, sess: &Session) {
|
||
|
|
let tf = InternedString::new("target_feature");
|
||
|
|
macro_rules! fillout {
|
||
|
|
($($func: ident, $name: expr;)*) => {{
|
||
|
|
$(if $func(sess) {
|
||
|
|
cfg.push(attr::mk_name_value_item_str(tf.clone(), intern($name)))
|
||
|
|
})*
|
||
|
|
}}
|
||
|
|
}
|
||
|
|
fillout! {
|
||
|
|
has_sse, "sse";
|
||
|
|
has_sse2, "sse2";
|
||
|
|
has_sse3, "sse3";
|
||
|
|
has_ssse3, "ssse3";
|
||
|
|
has_sse41, "sse4.1";
|
||
|
|
has_sse42, "sse4.2";
|
||
|
|
has_avx, "avx";
|
||
|
|
has_avx2, "avx2";
|
||
|
|
has_neon, "neon";
|
||
|
|
has_vfp, "vfp";
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
fn features_contain(sess: &Session, s: &str) -> bool {
|
||
|
|
sess.target.target.options.features.contains(s) ||
|
||
|
|
sess.opts.cg.target_feature.contains(s)
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn has_sse(sess: &Session) -> bool {
|
||
|
|
features_contain(sess, "+sse") ||
|
||
|
|
has_sse2(sess)
|
||
|
|
}
|
||
|
|
pub fn has_sse2(sess: &Session) -> bool {
|
||
|
|
// x86-64 requires at least SSE2 support
|
||
|
|
sess.target.target.arch == "x86_64" ||
|
||
|
|
features_contain(sess, "+sse2") ||
|
||
|
|
has_sse3(sess)
|
||
|
|
}
|
||
|
|
pub fn has_sse3(sess: &Session) -> bool {
|
||
|
|
features_contain(sess, "+sse3") ||
|
||
|
|
has_ssse3(sess)
|
||
|
|
}
|
||
|
|
pub fn has_ssse3(sess: &Session) -> bool {
|
||
|
|
features_contain(sess, "+ssse3") ||
|
||
|
|
has_sse41(sess)
|
||
|
|
}
|
||
|
|
pub fn has_sse41(sess: &Session) -> bool {
|
||
|
|
features_contain(sess, "+sse4.1") ||
|
||
|
|
has_sse42(sess)
|
||
|
|
}
|
||
|
|
pub fn has_sse42(sess: &Session) -> bool {
|
||
|
|
features_contain(sess, "+sse4.2") ||
|
||
|
|
has_avx(sess)
|
||
|
|
}
|
||
|
|
pub fn has_avx(sess: &Session) -> bool {
|
||
|
|
features_contain(sess, "+avx") ||
|
||
|
|
has_avx2(sess)
|
||
|
|
}
|
||
|
|
pub fn has_avx2(sess: &Session) -> bool {
|
||
|
|
features_contain(sess, "+avx2")
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn has_neon(sess: &Session) -> bool {
|
||
|
|
// AArch64 requires NEON support
|
||
|
|
sess.target.target.arch == "aarch64" ||
|
||
|
|
features_contain(sess, "+neon")
|
||
|
|
}
|
||
|
|
pub fn has_vfp(sess: &Session) -> bool {
|
||
|
|
// AArch64 requires VFP support
|
||
|
|
sess.target.target.arch == "aarch64" ||
|
||
|
|
features_contain(sess, "+vfp")
|
||
|
|
}
|