Skip to content

Commit 1cafa03

Browse files
kleisaukekleisauke
committed
Fix CI for targets that conditionally disable f16 or f128 support
1 parent 6a80b90 commit 1cafa03

17 files changed

+259
-182
lines changed

‎build.rs

+2-81
Original file line numberDiff line numberDiff line change
@@ -1,44 +1,8 @@
11
use std::{collections::BTreeMap, env, path::PathBuf, sync::atomic::Ordering};
22

3-
#[allow(dead_code)]
4-
struct Target {
5-
triple: String,
6-
os: String,
7-
arch: String,
8-
vendor: String,
9-
env: String,
10-
pointer_width: u8,
11-
little_endian: bool,
12-
features: Vec<String>,
13-
}
14-
15-
impl Target {
16-
fn from_env() -> Self {
17-
let little_endian = match env::var("CARGO_CFG_TARGET_ENDIAN").unwrap().as_str() {
18-
"little" => true,
19-
"big" => false,
20-
x => panic!("unknown endian {x}"),
21-
};
3+
mod configure;
224

23-
Self {
24-
triple: env::var("TARGET").unwrap(),
25-
os: env::var("CARGO_CFG_TARGET_OS").unwrap(),
26-
arch: env::var("CARGO_CFG_TARGET_ARCH").unwrap(),
27-
vendor: env::var("CARGO_CFG_TARGET_VENDOR").unwrap(),
28-
env: env::var("CARGO_CFG_TARGET_ENV").unwrap(),
29-
pointer_width: env::var("CARGO_CFG_TARGET_POINTER_WIDTH")
30-
.unwrap()
31-
.parse()
32-
.unwrap(),
33-
little_endian,
34-
features: env::var("CARGO_CFG_TARGET_FEATURE")
35-
.unwrap_or_default()
36-
.split(",")
37-
.map(ToOwned::to_owned)
38-
.collect(),
39-
}
40-
}
41-
}
5+
use configure::{configure_f16_f128, Target};
426

437
fn main() {
448
println!("cargo:rerun-if-changed=build.rs");
@@ -261,49 +225,6 @@ fn configure_check_cfg() {
261225
println!("cargo::rustc-check-cfg=cfg(assert_no_panic)");
262226
}
263227

264-
/// Configure whether or not `f16` and `f128` support should be enabled.
265-
fn configure_f16_f128(target: &Target) {
266-
// Set whether or not `f16` and `f128` are supported at a basic level by LLVM. This only means
267-
// that the backend will not crash when using these types. This does not mean that the
268-
// backend does the right thing, or that the platform doesn't have ABI bugs.
269-
//
270-
// We do this here rather than in `rust-lang/rust` because configuring via cargo features is
271-
// not straightforward.
272-
//
273-
// Original source of this list:
274-
// <https://github.com/rust-lang/compiler-builtins/pull/652#issuecomment-2266151350>
275-
let (f16_ok, f128_ok) = match target.arch.as_str() {
276-
// `f16` and `f128` both crash <https://github.com/llvm/llvm-project/issues/94434>
277-
"arm64ec" => (false, false),
278-
// `f16` crashes <https://github.com/llvm/llvm-project/issues/50374>
279-
"s390x" => (false, true),
280-
// `f128` crashes <https://github.com/llvm/llvm-project/issues/96432>
281-
"mips64" | "mips64r6" => (true, false),
282-
// `f128` crashes <https://github.com/llvm/llvm-project/issues/101545>
283-
"powerpc64" if &target.os == "aix" => (true, false),
284-
// `f128` crashes <https://github.com/llvm/llvm-project/issues/41838>
285-
"sparc" | "sparcv9" => (true, false),
286-
// `f16` miscompiles <https://github.com/llvm/llvm-project/issues/96438>
287-
"wasm32" | "wasm64" => (false, true),
288-
// Most everything else works as of LLVM 19
289-
_ => (true, true),
290-
};
291-
292-
// If the feature is set, disable these types.
293-
let disable_both = env::var_os("CARGO_FEATURE_NO_F16_F128").is_some();
294-
295-
println!("cargo::rustc-check-cfg=cfg(f16_enabled)");
296-
println!("cargo::rustc-check-cfg=cfg(f128_enabled)");
297-
298-
if f16_ok && !disable_both {
299-
println!("cargo::rustc-cfg=f16_enabled");
300-
}
301-
302-
if f128_ok && !disable_both {
303-
println!("cargo::rustc-cfg=f128_enabled");
304-
}
305-
}
306-
307228
#[cfg(feature = "c")]
308229
mod c {
309230
use std::collections::{BTreeMap, HashSet};

‎configure.rs

+85
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,85 @@
1+
/// Configuration that is shared between `compiler_builtins` and `testcrate`.
2+
use std::env;
3+
4+
#[allow(dead_code)]
5+
pub struct Target {
6+
pub triple: String,
7+
pub os: String,
8+
pub arch: String,
9+
pub vendor: String,
10+
pub env: String,
11+
pub pointer_width: u8,
12+
pub little_endian: bool,
13+
pub features: Vec<String>,
14+
}
15+
16+
impl Target {
17+
pub fn from_env() -> Self {
18+
let little_endian = match env::var("CARGO_CFG_TARGET_ENDIAN").unwrap().as_str() {
19+
"little" => true,
20+
"big" => false,
21+
x => panic!("unknown endian {x}"),
22+
};
23+
24+
Self {
25+
triple: env::var("TARGET").unwrap(),
26+
os: env::var("CARGO_CFG_TARGET_OS").unwrap(),
27+
arch: env::var("CARGO_CFG_TARGET_ARCH").unwrap(),
28+
vendor: env::var("CARGO_CFG_TARGET_VENDOR").unwrap(),
29+
env: env::var("CARGO_CFG_TARGET_ENV").unwrap(),
30+
pointer_width: env::var("CARGO_CFG_TARGET_POINTER_WIDTH")
31+
.unwrap()
32+
.parse()
33+
.unwrap(),
34+
little_endian,
35+
features: env::var("CARGO_CFG_TARGET_FEATURE")
36+
.unwrap_or_default()
37+
.split(",")
38+
.map(ToOwned::to_owned)
39+
.collect(),
40+
}
41+
}
42+
}
43+
44+
/// Configure whether or not `f16` and `f128` support should be enabled.
45+
pub fn configure_f16_f128(target: &Target) {
46+
// Set whether or not `f16` and `f128` are supported at a basic level by LLVM. This only means
47+
// that the backend will not crash when using these types. This does not mean that the
48+
// backend does the right thing, or that the platform doesn't have ABI bugs.
49+
//
50+
// We do this here rather than in `rust-lang/rust` because configuring via cargo features is
51+
// not straightforward.
52+
//
53+
// Original source of this list:
54+
// <https://github.com/rust-lang/compiler-builtins/pull/652#issuecomment-2266151350>
55+
let (f16_ok, f128_ok) = match target.arch.as_str() {
56+
// `f16` and `f128` both crash <https://github.com/llvm/llvm-project/issues/94434>
57+
"arm64ec" => (false, false),
58+
// `f16` crashes <https://github.com/llvm/llvm-project/issues/50374>
59+
"s390x" => (false, true),
60+
// `f128` crashes <https://github.com/llvm/llvm-project/issues/96432>
61+
"mips64" | "mips64r6" => (true, false),
62+
// `f128` crashes <https://github.com/llvm/llvm-project/issues/101545>
63+
"powerpc64" if &target.os == "aix" => (true, false),
64+
// `f128` crashes <https://github.com/llvm/llvm-project/issues/41838>
65+
"sparc" | "sparcv9" => (true, false),
66+
// `f16` miscompiles <https://github.com/llvm/llvm-project/issues/96438>
67+
"wasm32" | "wasm64" => (false, true),
68+
// Most everything else works as of LLVM 19
69+
_ => (true, true),
70+
};
71+
72+
// If the feature is set, disable these types.
73+
let disable_both = env::var_os("CARGO_FEATURE_NO_F16_F128").is_some();
74+
75+
println!("cargo::rustc-check-cfg=cfg(f16_enabled)");
76+
println!("cargo::rustc-check-cfg=cfg(f128_enabled)");
77+
78+
if f16_ok && !disable_both {
79+
println!("cargo::rustc-cfg=f16_enabled");
80+
}
81+
82+
if f128_ok && !disable_both {
83+
println!("cargo::rustc-cfg=f128_enabled");
84+
}
85+
}

‎testcrate/benches/float_add.rs

+15-3
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
1-
#![feature(f128)]
1+
#![cfg_attr(f128_enabled, feature(f128))]
22

33
use compiler_builtins::float::add;
4-
use criterion::{criterion_group, criterion_main, Criterion};
4+
use criterion::{criterion_main, Criterion};
55
use testcrate::float_bench;
66

77
float_bench! {
@@ -66,6 +66,7 @@ float_bench! {
6666
],
6767
}
6868

69+
#[cfg(f128_enabled)]
6970
float_bench! {
7071
name: add_f128,
7172
sig: (a: f128, b: f128) -> f128,
@@ -77,5 +78,16 @@ float_bench! {
7778
asm: []
7879
}
7980

80-
criterion_group!(float_add, add_f32, add_f64, add_f128);
81+
pub fn float_add() {
82+
let mut criterion = Criterion::default().configure_from_args();
83+
84+
add_f32(&mut criterion);
85+
add_f64(&mut criterion);
86+
87+
#[cfg(f128_enabled)]
88+
{
89+
add_f128(&mut criterion);
90+
}
91+
}
92+
8193
criterion_main!(float_add);

‎testcrate/benches/float_cmp.rs

+17-11
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
1-
#![feature(f128)]
1+
#![cfg_attr(f128_enabled, feature(f128))]
22

3-
use criterion::{criterion_group, criterion_main, Criterion};
3+
use criterion::{criterion_main, Criterion};
44
use testcrate::float_bench;
55

66
use compiler_builtins::float::cmp;
@@ -190,13 +190,19 @@ float_bench! {
190190
asm: []
191191
}
192192

193-
criterion_group!(
194-
float_cmp,
195-
cmp_f32_gt,
196-
cmp_f32_unord,
197-
cmp_f64_gt,
198-
cmp_f64_unord,
199-
cmp_f128_gt,
200-
cmp_f128_unord
201-
);
193+
pub fn float_cmp() {
194+
let mut criterion = Criterion::default().configure_from_args();
195+
196+
cmp_f32_gt(&mut criterion);
197+
cmp_f32_unord(&mut criterion);
198+
cmp_f64_gt(&mut criterion);
199+
cmp_f64_unord(&mut criterion);
200+
201+
#[cfg(f128_enabled)]
202+
{
203+
cmp_f128_gt(&mut criterion);
204+
cmp_f128_unord(&mut criterion);
205+
}
206+
}
207+
202208
criterion_main!(float_cmp);

‎testcrate/benches/float_conv.rs

-1
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
#![feature(f128)]
21
#![allow(improper_ctypes)]
32

43
use compiler_builtins::float::conv;

‎testcrate/benches/float_div.rs

-2
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,3 @@
1-
#![feature(f128)]
2-
31
use compiler_builtins::float::div;
42
use criterion::{criterion_group, criterion_main, Criterion};
53
use testcrate::float_bench;

‎testcrate/benches/float_extend.rs

+29-20
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,12 @@
11
#![allow(unused_variables)] // "unused" f16 registers
2-
#![feature(f128)]
3-
#![feature(f16)]
2+
#![cfg_attr(f128_enabled, feature(f128))]
3+
#![cfg_attr(f16_enabled, feature(f16))]
44

55
use compiler_builtins::float::extend;
6-
use criterion::{criterion_group, criterion_main, Criterion};
6+
use criterion::{criterion_main, Criterion};
77
use testcrate::float_bench;
88

9+
#[cfg(f16_enabled)]
910
float_bench! {
1011
name: extend_f16_f32,
1112
sig: (a: f16) -> f32,
@@ -28,6 +29,7 @@ float_bench! {
2829
],
2930
}
3031

32+
#[cfg(all(f16_enabled, f128_enabled))]
3133
float_bench! {
3234
name: extend_f16_f128,
3335
sig: (a: f16) -> f128,
@@ -60,6 +62,7 @@ float_bench! {
6062
],
6163
}
6264

65+
#[cfg(f128_enabled)]
6366
float_bench! {
6467
name: extend_f32_f128,
6568
sig: (a: f32) -> f128,
@@ -71,6 +74,7 @@ float_bench! {
7174
asm: [],
7275
}
7376

77+
#[cfg(f128_enabled)]
7478
float_bench! {
7579
name: extend_f64_f128,
7680
sig: (a: f64) -> f128,
@@ -82,23 +86,28 @@ float_bench! {
8286
asm: [],
8387
}
8488

85-
#[cfg(not(any(target_arch = "powerpc", target_arch = "powerpc64")))]
86-
criterion_group!(
87-
float_extend,
88-
extend_f16_f32,
89-
extend_f16_f128,
90-
extend_f32_f64,
91-
extend_f32_f128,
92-
extend_f64_f128,
93-
);
89+
pub fn float_extend() {
90+
let mut criterion = Criterion::default().configure_from_args();
9491

95-
// FIXME(#655): `f16` tests disabled until we can bootstrap symbols
96-
#[cfg(any(target_arch = "powerpc", target_arch = "powerpc64"))]
97-
criterion_group!(
98-
float_extend,
99-
extend_f32_f64,
100-
extend_f32_f128,
101-
extend_f64_f128,
102-
);
92+
// FIXME(#655): `f16` tests disabled until we can bootstrap symbols
93+
#[cfg(f16_enabled)]
94+
#[cfg(not(any(target_arch = "powerpc", target_arch = "powerpc64")))]
95+
{
96+
extend_f16_f32(&mut criterion);
97+
98+
#[cfg(f128_enabled)]
99+
{
100+
extend_f16_f128(&mut criterion);
101+
}
102+
}
103+
104+
extend_f32_f64(&mut criterion);
105+
106+
#[cfg(f128_enabled)]
107+
{
108+
extend_f32_f128(&mut criterion);
109+
extend_f64_f128(&mut criterion);
110+
}
111+
}
103112

104113
criterion_main!(float_extend);

‎testcrate/benches/float_mul.rs

+15-3
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
1-
#![feature(f128)]
1+
#![cfg_attr(f128_enabled, feature(f128))]
22

33
use compiler_builtins::float::mul;
4-
use criterion::{criterion_group, criterion_main, Criterion};
4+
use criterion::{criterion_main, Criterion};
55
use testcrate::float_bench;
66

77
float_bench! {
@@ -66,6 +66,7 @@ float_bench! {
6666
],
6767
}
6868

69+
#[cfg(f128_enabled)]
6970
float_bench! {
7071
name: mul_f128,
7172
sig: (a: f128, b: f128) -> f128,
@@ -77,5 +78,16 @@ float_bench! {
7778
asm: []
7879
}
7980

80-
criterion_group!(float_mul, mul_f32, mul_f64, mul_f128);
81+
pub fn float_mul() {
82+
let mut criterion = Criterion::default().configure_from_args();
83+
84+
mul_f32(&mut criterion);
85+
mul_f64(&mut criterion);
86+
87+
#[cfg(f128_enabled)]
88+
{
89+
mul_f128(&mut criterion);
90+
}
91+
}
92+
8193
criterion_main!(float_mul);

0 commit comments

Comments
 (0)