Skip to content

Commit cfadfab

Browse files
andjo403andjo403
committed
Add range attribute to scalar function results and arguments
1 parent 9bad7ba commit cfadfab

File tree

12 files changed

+159
-24
lines changed

12 files changed

+159
-24
lines changed

‎compiler/rustc_codegen_llvm/src/abi.rs

Lines changed: 48 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -415,9 +415,32 @@ impl<'ll, 'tcx> FnAbiLlvmExt<'ll, 'tcx> for FnAbi<'tcx, Ty<'tcx>> {
415415
i += 1;
416416
i - 1
417417
};
418+
419+
let apply_range_attr = |idx: AttributePlace, scalar: rustc_target::abi::Scalar| {
420+
if cx.sess().opts.optimize != config::OptLevel::No
421+
&& llvm_util::get_version() >= (19, 0, 0)
422+
&& matches!(scalar.primitive(), Int(..))
423+
// If the value is a boolean, the range is 0..2 and that ultimately
424+
// become 0..0 when the type becomes i1, which would be rejected
425+
// by the LLVM verifier.
426+
&& !scalar.is_bool()
427+
// LLVM also rejects full range.
428+
&& !scalar.is_always_valid(cx)
429+
{
430+
attributes::apply_to_llfn(
431+
llfn,
432+
idx,
433+
&[llvm::CreateRangeAttr(cx.llcx, scalar.size(cx), scalar.valid_range(cx))],
434+
);
435+
}
436+
};
437+
418438
match &self.ret.mode {
419439
PassMode::Direct(attrs) => {
420440
attrs.apply_attrs_to_llfn(llvm::AttributePlace::ReturnValue, cx, llfn);
441+
if let abi::Abi::Scalar(scalar) = self.ret.layout.abi {
442+
apply_range_attr(llvm::AttributePlace::ReturnValue, scalar);
443+
}
421444
}
422445
PassMode::Indirect { attrs, meta_attrs: _, on_stack } => {
423446
assert!(!on_stack);
@@ -456,8 +479,13 @@ impl<'ll, 'tcx> FnAbiLlvmExt<'ll, 'tcx> for FnAbi<'tcx, Ty<'tcx>> {
456479
);
457480
attributes::apply_to_llfn(llfn, llvm::AttributePlace::Argument(i), &[byval]);
458481
}
459-
PassMode::Direct(attrs)
460-
| PassMode::Indirect { attrs, meta_attrs: None, on_stack: false } => {
482+
PassMode::Direct(attrs) => {
483+
let i = apply(attrs);
484+
if let abi::Abi::Scalar(scalar) = arg.layout.abi {
485+
apply_range_attr(llvm::AttributePlace::Argument(i), scalar);
486+
}
487+
}
488+
PassMode::Indirect { attrs, meta_attrs: None, on_stack: false } => {
461489
apply(attrs);
462490
}
463491
PassMode::Indirect { attrs, meta_attrs: Some(meta_attrs), on_stack } => {
@@ -466,8 +494,12 @@ impl<'ll, 'tcx> FnAbiLlvmExt<'ll, 'tcx> for FnAbi<'tcx, Ty<'tcx>> {
466494
apply(meta_attrs);
467495
}
468496
PassMode::Pair(a, b) => {
469-
apply(a);
470-
apply(b);
497+
let i = apply(a);
498+
let ii = apply(b);
499+
if let abi::Abi::ScalarPair(scalar_a, scalar_b) = arg.layout.abi {
500+
apply_range_attr(llvm::AttributePlace::Argument(i), scalar_a);
501+
apply_range_attr(llvm::AttributePlace::Argument(ii), scalar_b);
502+
}
471503
}
472504
PassMode::Cast { cast, pad_i32 } => {
473505
if *pad_i32 {
@@ -517,15 +549,18 @@ impl<'ll, 'tcx> FnAbiLlvmExt<'ll, 'tcx> for FnAbi<'tcx, Ty<'tcx>> {
517549
}
518550
_ => {}
519551
}
520-
if let abi::Abi::Scalar(scalar) = self.ret.layout.abi {
521-
// If the value is a boolean, the range is 0..2 and that ultimately
522-
// become 0..0 when the type becomes i1, which would be rejected
523-
// by the LLVM verifier.
524-
if let Int(..) = scalar.primitive() {
525-
if !scalar.is_bool() && !scalar.is_always_valid(bx) {
526-
bx.range_metadata(callsite, scalar.valid_range(bx));
527-
}
528-
}
552+
if bx.cx.sess().opts.optimize != config::OptLevel::No
553+
&& llvm_util::get_version() < (19, 0, 0)
554+
&& let abi::Abi::Scalar(scalar) = self.ret.layout.abi
555+
&& matches!(scalar.primitive(), Int(..))
556+
// If the value is a boolean, the range is 0..2 and that ultimately
557+
// become 0..0 when the type becomes i1, which would be rejected
558+
// by the LLVM verifier.
559+
&& !scalar.is_bool()
560+
// LLVM also rejects full range.
561+
&& !scalar.is_always_valid(bx)
562+
{
563+
bx.range_metadata(callsite, scalar.valid_range(bx));
529564
}
530565
for arg in self.args.iter() {
531566
match &arg.mode {

‎compiler/rustc_codegen_llvm/src/llvm/ffi.rs

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1575,6 +1575,12 @@ extern "C" {
15751575
pub fn LLVMRustCreateAllocSizeAttr(C: &Context, size_arg: u32) -> &Attribute;
15761576
pub fn LLVMRustCreateAllocKindAttr(C: &Context, size_arg: u64) -> &Attribute;
15771577
pub fn LLVMRustCreateMemoryEffectsAttr(C: &Context, effects: MemoryEffects) -> &Attribute;
1578+
pub fn LLVMRustCreateRangeAttribute(
1579+
C: &Context,
1580+
num_bits: c_uint,
1581+
lower_words: *const u64,
1582+
upper_words: *const u64,
1583+
) -> &Attribute;
15781584

15791585
// Operations on functions
15801586
pub fn LLVMRustGetOrInsertFunction<'a>(

‎compiler/rustc_codegen_llvm/src/llvm/mod.rs

Lines changed: 16 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ use std::string::FromUtf8Error;
88
use libc::c_uint;
99
use rustc_data_structures::small_c_str::SmallCStr;
1010
use rustc_llvm::RustString;
11-
use rustc_target::abi::Align;
11+
use rustc_target::abi::{Align, Size, WrappingRange};
1212

1313
pub use self::AtomicRmwBinOp::*;
1414
pub use self::CallConv::*;
@@ -105,6 +105,21 @@ pub fn CreateAllocKindAttr(llcx: &Context, kind_arg: AllocKindFlags) -> &Attribu
105105
unsafe { LLVMRustCreateAllocKindAttr(llcx, kind_arg.bits()) }
106106
}
107107

108+
pub fn CreateRangeAttr(llcx: &Context, size: Size, range: WrappingRange) -> &Attribute {
109+
let lower = range.start;
110+
let upper = range.end.wrapping_add(1);
111+
let lower_words = [lower as u64, (lower >> 64) as u64];
112+
let upper_words = [upper as u64, (upper >> 64) as u64];
113+
unsafe {
114+
LLVMRustCreateRangeAttribute(
115+
llcx,
116+
size.bits().try_into().unwrap(),
117+
lower_words.as_ptr(),
118+
upper_words.as_ptr(),
119+
)
120+
}
121+
}
122+
108123
#[derive(Copy, Clone)]
109124
pub enum AttributePlace {
110125
ReturnValue,

‎compiler/rustc_llvm/llvm-wrapper/RustWrapper.cpp

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -397,6 +397,18 @@ LLVMRustCreateAllocSizeAttr(LLVMContextRef C, uint32_t ElementSizeArg) {
397397
std::nullopt));
398398
}
399399

400+
extern "C" LLVMAttributeRef
401+
LLVMRustCreateRangeAttribute(LLVMContextRef C, unsigned NumBits,
402+
const uint64_t LowerWords[],
403+
const uint64_t UpperWords[]) {
404+
#if LLVM_VERSION_GE(19, 0)
405+
return LLVMCreateConstantRangeAttribute(C, Attribute::Range, NumBits,
406+
LowerWords, UpperWords);
407+
#else
408+
report_fatal_error("LLVM 19.0 is required for Range Attribute");
409+
#endif
410+
}
411+
400412
// These values **must** match ffi::AllocKindFlags.
401413
// It _happens_ to match the LLVM values of llvm::AllocFnKind,
402414
// but that's happenstance and we do explicit conversions before

‎tests/codegen/call-metadata.rs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
// scalar value.
33

44
//@ compile-flags: -O -C no-prepopulate-passes
5+
//@ ignore-llvm-version: 19 - 99
56

67
#![crate_type = "lib"]
78

‎tests/codegen/cast-optimized.rs

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,8 +20,6 @@ pub fn u32_index(c: u32) -> [bool; 22] {
2020
// CHECK-LABEL: @char_as_u32_index
2121
#[no_mangle]
2222
pub fn char_as_u32_index(c: char) -> [bool; 22] {
23-
// CHECK: %[[B:.+]] = icmp ult i32 %c, 1114112
24-
// CHECK: call void @llvm.assume(i1 %[[B]])
2523
let c = c as u32;
2624

2725
let mut array = [false; 22];

‎tests/codegen/common_prim_int_ptr.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ pub fn insert_box(x: Box<()>) -> Result<usize, Box<()>> {
2828

2929
// CHECK-LABEL: @extract_int
3030
// CHECK-NOT: nonnull
31-
// CHECK-SAME: (i{{[0-9]+}} {{[^,]+}} [[DISCRIMINANT:%[0-9]+]], ptr {{[^,]+}} [[PAYLOAD:%[0-9]+]])
31+
// CHECK-SAME: (i{{[0-9]+}} {{[^%]+}} [[DISCRIMINANT:%[0-9]+]], ptr {{[^,]+}} [[PAYLOAD:%[0-9]+]])
3232
#[no_mangle]
3333
pub unsafe fn extract_int(x: Result<usize, Box<()>>) -> usize {
3434
// CHECK: [[TEMP:%.+]] = ptrtoint ptr [[PAYLOAD]] to [[USIZE:i[0-9]+]]
@@ -40,7 +40,7 @@ pub unsafe fn extract_int(x: Result<usize, Box<()>>) -> usize {
4040
}
4141

4242
// CHECK-LABEL: @extract_box
43-
// CHECK-SAME: (i{{[0-9]+}} {{[^,]+}} [[DISCRIMINANT:%[0-9]+]], ptr {{[^,]+}} [[PAYLOAD:%[0-9]+]])
43+
// CHECK-SAME: (i{{[0-9]+}} {{[^%]+}} [[DISCRIMINANT:%[0-9]+]], ptr {{[^,]+}} [[PAYLOAD:%[0-9]+]])
4444
#[no_mangle]
4545
pub unsafe fn extract_box(x: Result<usize, Box<i32>>) -> Box<i32> {
4646
// CHECK: ret ptr [[PAYLOAD]]

‎tests/codegen/enum/enum-match.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ pub enum Enum1 {
3434

3535
// CHECK: define noundef{{( range\(i8 [0-9]+, [0-9]+\))?}} i8 @match1{{.*}}
3636
// CHECK-NEXT: start:
37-
// CHECK-NEXT: %1 = add i8 %0, -2
37+
// CHECK-NEXT: %1 = add{{( nsw)?}} i8 %0, -2
3838
// CHECK-NEXT: %2 = zext i8 %1 to i64
3939
// CHECK-NEXT: %3 = icmp ult i8 %1, 2
4040
// CHECK-NEXT: %4 = add nuw nsw i64 %2, 1

‎tests/codegen/function-arguments.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ pub fn maybeuninit_enum_bool(x: MaybeUninit<MyBool>) -> MaybeUninit<MyBool> {
5050
x
5151
}
5252

53-
// CHECK: noundef i32 @char(i32 noundef %x)
53+
// CHECK: noundef{{( range\(i32 0, 1114112\))?}} i32 @char(i32 noundef{{( range\(i32 0, 1114112\))?}} %x)
5454
#[no_mangle]
5555
pub fn char(x: char) -> char {
5656
x
@@ -68,7 +68,7 @@ pub fn int(x: u64) -> u64 {
6868
x
6969
}
7070

71-
// CHECK: noundef i64 @nonzero_int(i64 noundef %x)
71+
// CHECK: noundef{{( range\(i64 1, 0\))?}} i64 @nonzero_int(i64 noundef{{( range\(i64 1, 0\))?}} %x)
7272
#[no_mangle]
7373
pub fn nonzero_int(x: NonZero<u64>) -> NonZero<u64> {
7474
x
@@ -250,7 +250,7 @@ pub fn return_slice(x: &[u16]) -> &[u16] {
250250
x
251251
}
252252

253-
// CHECK: { i16, i16 } @enum_id_1(i16 noundef %x.0, i16 %x.1)
253+
// CHECK: { i16, i16 } @enum_id_1(i16 noundef{{( range\(i16 0, 3\))?}} %x.0, i16 %x.1)
254254
#[no_mangle]
255255
pub fn enum_id_1(x: Option<Result<u16, u16>>) -> Option<Result<u16, u16>> {
256256
x

‎tests/codegen/issues/issue-68667-unwrap-combinators.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
// MIR inlining now optimizes this code.
66

77
// CHECK-LABEL: @unwrap_combinators
8-
// CHECK: icmp
8+
// CHECK: {{icmp|trunc}}
99
// CHECK-NEXT: icmp
1010
// CHECK-NEXT: select i1
1111
// CHECK-NEXT: ret i1

0 commit comments

Comments
 (0)