llvm-project/llvm/test/CodeGen/BPF/adjust-opt-icmp3.ll
Yonghong Song f63405f6e3 BPF: Workaround an InstCombine ICmp transformation with llvm.bpf.compare builtin
Commit acabad9ff6bf ("[InstCombine] try to canonicalize icmp with
trunc op into mask and cmp") added a transformation to
convert "(conv)a < power_2_const" to "a & <const>" in certain
cases and bpf kernel verifier has to handle the resulted code
conservatively and this may reject otherwise legitimate program.

This commit tries to prevent such a transformation. A bpf backend
builtin llvm.bpf.compare is added. The ICMP insn, which is subject to
above InstCombine transformation, is converted to the builtin
function. The builtin function is later lowered to original ICMP insn,
certainly after InstCombine pass.

With this change, all affected bpf strobemeta* selftests are
passed now.

Differential Revision: https://reviews.llvm.org/D112938
2021-11-01 14:46:20 -07:00

86 lines
2.6 KiB
LLVM

; RUN: opt -O2 -S -mtriple=bpf-pc-linux %s -o %t1
; RUN: llc %t1 -o - | FileCheck -check-prefixes=CHECK,CHECK-V1 %s
; RUN: opt -O2 -S -mtriple=bpf-pc-linux %s -o %t1
; RUN: llc %t1 -mcpu=v3 -o - | FileCheck -check-prefixes=CHECK,CHECK-V3 %s
;
; Source:
; int test1(unsigned long a) {
; if ((unsigned)a <= 3) return 2;
; return 3;
; }
; int test2(unsigned long a) {
; if ((unsigned)a < 4) return 2;
; return 3;
; }
; Compilation flag:
; clang -target bpf -O2 -S -emit-llvm -Xclang -disable-llvm-passes test.c
; Function Attrs: nounwind
define dso_local i32 @test1(i64 %a) #0 {
entry:
%retval = alloca i32, align 4
%a.addr = alloca i64, align 8
store i64 %a, i64* %a.addr, align 8, !tbaa !3
%0 = load i64, i64* %a.addr, align 8, !tbaa !3
%conv = trunc i64 %0 to i32
%cmp = icmp ule i32 %conv, 3
br i1 %cmp, label %if.then, label %if.end
if.then: ; preds = %entry
store i32 2, i32* %retval, align 4
br label %return
if.end: ; preds = %entry
store i32 3, i32* %retval, align 4
br label %return
return: ; preds = %if.end, %if.then
%1 = load i32, i32* %retval, align 4
ret i32 %1
}
; CHECK-LABEL: test1
; CHECK-V1: if r[[#]] > r[[#]] goto
; CHECK-V3: if w[[#]] < 4 goto
; Function Attrs: nounwind
define dso_local i32 @test2(i64 %a) #0 {
entry:
%retval = alloca i32, align 4
%a.addr = alloca i64, align 8
store i64 %a, i64* %a.addr, align 8, !tbaa !3
%0 = load i64, i64* %a.addr, align 8, !tbaa !3
%conv = trunc i64 %0 to i32
%cmp = icmp ult i32 %conv, 4
br i1 %cmp, label %if.then, label %if.end
if.then: ; preds = %entry
store i32 2, i32* %retval, align 4
br label %return
if.end: ; preds = %entry
store i32 3, i32* %retval, align 4
br label %return
return: ; preds = %if.end, %if.then
%1 = load i32, i32* %retval, align 4
ret i32 %1
}
; CHECK-LABEL: test2
; CHECK-V1: if r[[#]] > r[[#]] goto
; CHECK-V3: if w[[#]] < 4 goto
attributes #0 = { nounwind "frame-pointer"="all" "min-legal-vector-width"="0" "no-trapping-math"="true" "stack-protector-buffer-size"="8" }
!llvm.module.flags = !{!0, !1}
!llvm.ident = !{!2}
!0 = !{i32 1, !"wchar_size", i32 4}
!1 = !{i32 7, !"frame-pointer", i32 2}
!2 = !{!"clang version 14.0.0 (https://github.com/llvm/llvm-project.git b7892f95881c891032742e0cd81861b845512653)"}
!3 = !{!4, !4, i64 0}
!4 = !{!"long", !5, i64 0}
!5 = !{!"omnipotent char", !6, i64 0}
!6 = !{!"Simple C/C++ TBAA"}