Skip to content

Commit 9b14af5

Browse files
mysterymathmysterymath
committed
Add test cases and polish change
1 parent 7f725c0 commit 9b14af5

File tree

5 files changed

+172
-16
lines changed

5 files changed

+172
-16
lines changed

‎llvm/docs/LangRef.rst‎

Lines changed: 10 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -2747,18 +2747,17 @@ For example:
27472747
all arguments are not undef and not poison. Otherwise, it is undefined
27482748
behavior.
27492749

2750-
``"modular-format"="<type>,<string_idx>,<first_idx_to_check>,<modular_impl_fn>,<impl_name>,<aspects...>"``
2750+
``"modular-format"="<type>,<string_idx>,<first_arg_idx>,<modular_impl_fn>,<impl_name>,<aspects...>"``
27512751
This attribute indicates that the implementation is modular on a particular
2752-
format string argument . When the argument for a given call is constant, the
2753-
compiler may redirect the call to a modular implementation function
2754-
instead.
2755-
2756-
The compiler also emits relocations to report various aspects of the format
2757-
string and arguments that were present. The compiler reports an aspect by
2758-
issuing a relocation for the symbol `<impl_name>_<aspect>``. This arranges
2759-
for code and data needed to support the aspect of the implementation to be
2760-
brought into the link to satisfy weak references in the modular
2761-
implemenation function.
2752+
format string argument. If the compiler can determine that not all aspects
2753+
of the implementation are needed, it can report which aspects were needed
2754+
and redirect the call to a modular implementation function instead.
2755+
2756+
The compiler reports that an implementation aspect is needed by issuing a
2757+
relocation for the symbol `<impl_name>_<aspect>``. This arranges for code
2758+
and data needed to support the aspect of the implementation to be brought
2759+
into the link to satisfy weak references in the modular implemenation
2760+
function.
27622761

27632762
The first three arguments have the same semantics as the arguments to the C
27642763
``format`` attribute.

‎llvm/lib/IR/Verifier.cpp‎

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2554,6 +2554,20 @@ void Verifier::verifyFunctionAttrs(FunctionType *FT, AttributeList Attrs,
25542554
CheckFailed("invalid value for 'denormal-fp-math-f32' attribute: " + S,
25552555
V);
25562556
}
2557+
2558+
if (auto A = Attrs.getFnAttr("modular-format"); A.isValid()) {
2559+
StringRef S = A.getValueAsString();
2560+
SmallVector<StringRef> Args;
2561+
S.split(Args, ',');
2562+
Check(Args.size() >= 5,
2563+
"modular-format attribute requires at least 5 arguments", V);
2564+
unsigned FirstArgIdx;
2565+
Check(!Args[2].getAsInteger(10, FirstArgIdx),
2566+
"modular-format attribute first arg index is not an integer", V);
2567+
unsigned UpperBound = FT->getNumParams() + (FT->isVarArg() ? 1 : 0);
2568+
Check(FirstArgIdx > 0 && FirstArgIdx <= UpperBound,
2569+
"modular-format attribute first arg index is out of bounds", V);
2570+
}
25572571
}
25582572
void Verifier::verifyUnknownProfileMetadata(MDNode *MD) {
25592573
Check(MD->getNumOperands() == 2,

‎llvm/lib/Transforms/InstCombine/InstCombineCalls.cpp‎

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -4099,12 +4099,10 @@ static Value *optimizeModularFormat(CallInst *CI, IRBuilderBase &B) {
40994099
SmallVector<StringRef> Args(
41004100
llvm::split(CI->getFnAttr("modular-format").getValueAsString(), ','));
41014101
// TODO: Make use of the first two arguments
4102-
// TODO: Error handling
41034102
unsigned FirstArgIdx;
4104-
if (!llvm::to_integer(Args[2], FirstArgIdx))
4105-
return nullptr;
4106-
if (FirstArgIdx == 0)
4107-
return nullptr;
4103+
[[maybe_unused]] bool Error;
4104+
Error = Args[2].getAsInteger(10, FirstArgIdx);
4105+
assert(!Error && "invalid first arg index");
41084106
--FirstArgIdx;
41094107
StringRef FnName = Args[3];
41104108
StringRef ImplName = Args[4];

‎llvm/test/Transforms/InstCombine/modular-format.ll‎

Lines changed: 104 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,104 @@
1+
; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
2+
; Test that the modular format string library call simplifier works correctly.
3+
;
4+
; RUN: opt < %s -passes=instcombine -S | FileCheck %s
5+
6+
target datalayout = "e-p:32:32:32-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:32:64-f32:32:32-f64:32:64-v64:64:64-v128:128:128-a0:0:64-f80:128:128"
7+
8+
@.str.int = constant [3 x i8] c"%d\00"
9+
@.str.float = constant [3 x i8] c"%f\00"
10+
@.str.multi = constant [6 x i8] c"%f %d\00"
11+
@.str.multifp = constant [6 x i8] c"%f %f\00"
12+
@.str.noargs = constant [1 x i8] c"\00"
13+
14+
; Basic Transformation
15+
define void @test_basic(i32 %arg) {
16+
; CHECK-LABEL: @test_basic(
17+
; CHECK-NEXT: call void (ptr, ...) @basic_mod(ptr nonnull @.str.int, i32 [[ARG:%.*]])
18+
; CHECK-NEXT: ret void
19+
;
20+
call void (ptr, ...) @basic(ptr @.str.int, i32 %arg)
21+
ret void
22+
}
23+
24+
declare void @basic(ptr, ...) "modular-format"="printf,1,2,basic_mod,basic_impl"
25+
; "float" Aspect - Present
26+
define void @test_float_present(double %arg) {
27+
; CHECK-LABEL: @test_float_present(
28+
; CHECK-NEXT: call void (ptr, ...) @float_present_mod(ptr nonnull @.str.float, double [[ARG:%.*]])
29+
; CHECK-NEXT: call void @llvm.reloc.none(ptr nonnull @basic_impl_float)
30+
; CHECK-NEXT: ret void
31+
;
32+
call void (ptr, ...) @float_present(ptr @.str.float, double %arg)
33+
ret void
34+
}
35+
36+
declare void @float_present(ptr, ...) #0
37+
38+
; Unknown Aspects
39+
define void @test_unknown_aspects(i32 %arg) {
40+
; CHECK-LABEL: @test_unknown_aspects(
41+
; CHECK-NEXT: call void (ptr, ...) @unknown_aspects_mod(ptr nonnull @.str.int, i32 [[ARG:%.*]])
42+
; CHECK-NEXT: call void @llvm.reloc.none(ptr nonnull @basic_impl_unknown1)
43+
; CHECK-NEXT: call void @llvm.reloc.none(ptr nonnull @basic_impl_unknown2)
44+
; CHECK-NEXT: ret void
45+
;
46+
call void (ptr, ...) @unknown_aspects(ptr @.str.int, i32 %arg)
47+
ret void
48+
}
49+
50+
declare void @unknown_aspects(ptr, ...) "modular-format"="printf,1,2,unknown_aspects_mod,basic_impl,unknown1,unknown2"
51+
52+
; Multiple Aspects
53+
define void @test_multiple_aspects(double %arg1, i32 %arg2) {
54+
; CHECK-LABEL: @test_multiple_aspects(
55+
; CHECK-NEXT: call void (ptr, ...) @multiple_aspects_mod(ptr nonnull @.str.multi, double [[ARG1:%.*]], i32 [[ARG2:%.*]])
56+
; CHECK-NEXT: call void @llvm.reloc.none(ptr nonnull @basic_impl_float)
57+
; CHECK-NEXT: call void @llvm.reloc.none(ptr nonnull @basic_impl_unknown)
58+
; CHECK-NEXT: ret void
59+
;
60+
call void (ptr, ...) @multiple_aspects(ptr @.str.multi, double %arg1, i32 %arg2)
61+
ret void
62+
}
63+
64+
declare void @multiple_aspects(ptr, ...) "modular-format"="printf,1,2,multiple_aspects_mod,basic_impl,float,unknown"
65+
66+
; Multiple Floating-Point Arguments
67+
define void @test_multiple_fp_args(double %arg1, float %arg2) {
68+
; CHECK-LABEL: @test_multiple_fp_args(
69+
; CHECK-NEXT: call void (ptr, ...) @float_present_mod(ptr nonnull @.str.multifp, double [[ARG1:%.*]], float [[ARG2:%.*]])
70+
; CHECK-NEXT: call void @llvm.reloc.none(ptr nonnull @basic_impl_float)
71+
; CHECK-NEXT: ret void
72+
;
73+
call void (ptr, ...) @multiple_fp_args(ptr @.str.multifp, double %arg1, float %arg2)
74+
ret void
75+
}
76+
77+
declare void @multiple_fp_args(ptr, ...) #0
78+
79+
; No Arguments to Check
80+
define void @test_no_args_to_check() {
81+
; CHECK-LABEL: @test_no_args_to_check(
82+
; CHECK-NEXT: call void (ptr, ...) @float_present_mod(ptr nonnull @.str.noargs)
83+
; CHECK-NEXT: ret void
84+
;
85+
call void (ptr, ...) @no_args_to_check(ptr @.str.noargs)
86+
ret void
87+
}
88+
89+
declare void @no_args_to_check(ptr, ...) #0
90+
91+
; First argument index != 2
92+
define void @test_first_arg_idx(i32 %ignored, double %arg) {
93+
; CHECK-LABEL: @test_first_arg_idx(
94+
; CHECK-NEXT: call void (i32, ptr, ...) @first_arg_idx_mod(i32 [[IGNORED:%.*]], ptr nonnull @.str.float, double [[ARG:%.*]])
95+
; CHECK-NEXT: call void @llvm.reloc.none(ptr nonnull @basic_impl_float)
96+
; CHECK-NEXT: ret void
97+
;
98+
call void (i32, ptr, ...) @first_arg_idx(i32 %ignored, ptr @.str.float, double %arg)
99+
ret void
100+
}
101+
102+
declare void @first_arg_idx(i32, ptr, ...) "modular-format"="printf,2,3,first_arg_idx_mod,basic_impl,float"
103+
104+
attributes #0 = { "modular-format"="printf,1,2,float_present_mod,basic_impl,float" }

‎llvm/test/Verifier/modular-format.ll‎

Lines changed: 41 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,41 @@
1+
; RUN: not llvm-as < %s -o /dev/null 2>&1 | FileCheck %s
2+
3+
define void @test_too_few_arguments(i32 %arg, ...) "modular-format"="printf,1,2,basic_mod" {
4+
ret void
5+
}
6+
; CHECK: modular-format attribute requires at least 5 arguments
7+
; CHECK-NEXT: ptr @test_too_few_arguments
8+
9+
define void @test_first_arg_index_not_integer(i32 %arg, ...) "modular-format"="printf,1,foo,basic_mod,basic_impl" {
10+
ret void
11+
}
12+
; CHECK: modular-format attribute first arg index is not an integer
13+
; CHECK-NEXT: ptr @test_first_arg_index_not_integer
14+
15+
define void @test_first_arg_index_zero(i32 %arg) "modular-format"="printf,1,0,basic_mod,basic_impl" {
16+
ret void
17+
}
18+
; CHECK: modular-format attribute first arg index is out of bounds
19+
; CHECK-NEXT: ptr @test_first_arg_index_zero
20+
21+
define void @test_first_arg_index_out_of_bounds(i32 %arg) "modular-format"="printf,1,2,basic_mod,basic_impl" {
22+
ret void
23+
}
24+
; CHECK: modular-format attribute first arg index is out of bounds
25+
; CHECK-NEXT: ptr @test_first_arg_index_out_of_bounds
26+
27+
define void @test_first_arg_index_out_of_bounds_varargs(i32 %arg, ...) "modular-format"="printf,1,3,basic_mod,basic_impl" {
28+
ret void
29+
}
30+
; CHECK: modular-format attribute first arg index is out of bounds
31+
; CHECK-NEXT: ptr @test_first_arg_index_out_of_bounds_varargs
32+
33+
; CHECK-NOT: ptr @test_first_arg_index_in_bounds
34+
define void @test_first_arg_index_in_bounds(i32 %arg) "modular-format"="printf,1,1,basic_mod,basic_impl" {
35+
ret void
36+
}
37+
38+
; CHECK-NOT: ptr @test_first_arg_index_in_bounds_varargs
39+
define void @test_first_arg_index_in_bounds_varargs(i32 %arg, ...) "modular-format"="printf,1,2,basic_mod,basic_impl" {
40+
ret void
41+
}

0 commit comments

Comments
 (0)