[llvm] a4e437e - [FuncSpec] Add a test for specialising on a non-constant global argument. NFC.

[Sjoerd Meijer via llvm-commits]

Author: Sjoerd Meijer
Date: 2021-09-16T09:17:39+01:00
New Revision: a4e437e3c959ac0cb2edb733d081edc95a4fff22

URL: https://github.com/llvm/llvm-project/commit/a4e437e3c959ac0cb2edb733d081edc95a4fff22
DIFF: https://github.com/llvm/llvm-project/commit/a4e437e3c959ac0cb2edb733d081edc95a4fff22.diff

LOG: [FuncSpec] Add a test for specialising on a non-constant global argument. NFC.

Added: 
    llvm/test/Transforms/FunctionSpecialization/function-specialization-nonconst-glob.ll

Modified: 
    

Removed: 
    


################################################################################
diff  --git a/llvm/test/Transforms/FunctionSpecialization/function-specialization-nonconst-glob.ll b/llvm/test/Transforms/FunctionSpecialization/function-specialization-nonconst-glob.ll
new file mode 100644
index 0000000000000..e039cfc02f7fb
--- /dev/null
+++ b/llvm/test/Transforms/FunctionSpecialization/function-specialization-nonconst-glob.ll
@@ -0,0 +1,69 @@
+; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
+; RUN: opt -function-specialization -force-function-specialization -S < %s | FileCheck %s
+
+; Global B is not constant. We do specialise on this global, showing we
+; specialise on its address.
+
+target datalayout = "e-m:e-i8:8:32-i16:16:32-i64:64-i128:128-n32:64-S128"
+
+ at A = external dso_local constant i32, align 4
+ at B = external dso_local global i32, align 4
+
+define dso_local i32 @bar(i32 %x, i32 %y) {
+; CHECK-LABEL: @bar(
+; CHECK-NEXT:  entry:
+; CHECK-NEXT:    [[TOBOOL:%.*]] = icmp ne i32 [[X:%.*]], 0
+; CHECK-NEXT:    br i1 [[TOBOOL]], label [[IF_THEN:%.*]], label [[IF_ELSE:%.*]]
+; CHECK:       if.then:
+; CHECK-NEXT:    [[CALL:%.*]] = call i32 @foo.1(i32 [[X]], i32* @A)
+; CHECK-NEXT:    br label [[RETURN:%.*]]
+; CHECK:       if.else:
+; CHECK-NEXT:    [[CALL1:%.*]] = call i32 @foo.2(i32 [[Y:%.*]], i32* @B)
+; CHECK-NEXT:    br label [[RETURN]]
+; CHECK:       return:
+; CHECK-NEXT:    [[RETVAL_0:%.*]] = phi i32 [ [[CALL]], [[IF_THEN]] ], [ [[CALL1]], [[IF_ELSE]] ]
+; CHECK-NEXT:    ret i32 [[RETVAL_0]]
+;
+entry:
+  %tobool = icmp ne i32 %x, 0
+  br i1 %tobool, label %if.then, label %if.else
+
+if.then:
+  %call = call i32 @foo(i32 %x, i32* @A)
+  br label %return
+
+if.else:
+  %call1 = call i32 @foo(i32 %y, i32* @B)
+  br label %return
+
+return:
+  %retval.0 = phi i32 [ %call, %if.then ], [ %call1, %if.else ]
+  ret i32 %retval.0
+}
+
+define internal i32 @foo(i32 %x, i32* %b) {
+; CHECK-LABEL: @foo(
+; CHECK-NEXT:  entry:
+; CHECK-NEXT:    [[TMP0:%.*]] = load i32, i32* [[B:%.*]], align 4
+; CHECK-NEXT:    [[ADD:%.*]] = add nsw i32 [[X:%.*]], [[TMP0]]
+; CHECK-NEXT:    ret i32 [[ADD]]
+;
+entry:
+  %0 = load i32, i32* %b, align 4
+  %add = add nsw i32 %x, %0
+  ret i32 %add
+}
+
+; CHECK-LABEL: define internal i32 @foo.1(i32 %x, i32* %b) {
+; CHECK-NEXT:  entry:
+; CHECK-NEXT:    %0 = load i32, i32* @A, align 4
+; CHECK-NEXT:    %add = add nsw i32 %x, %0
+; CHECK-NEXT:    ret i32 %add
+; CHECK-NEXT:  }
+
+; CHECK-LABEL: define internal i32 @foo.2(i32 %x, i32* %b) {
+; CHECK-NEXT:  entry:
+; CHECK-NEXT:    %0 = load i32, i32* @B, align 4
+; CHECK-NEXT:    %add = add nsw i32 %x, %0
+; CHECK-NEXT:    ret i32 %add
+; CHECK-NEXT:  }