llvm/test/Transforms/FunctionSpecialization/function-specialization2.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
   2 ; RUN: opt -passes="ipsccp<func-spec>,deadargelim" -force-specialization -S < %s | FileCheck %s
   3 ; RUN: opt -passes="ipsccp<func-spec>,deadargelim" -funcspec-max-iters=1 -force-specialization -S < %s | FileCheck %s
   4 ; RUN: opt -passes="ipsccp<func-spec>,deadargelim" -funcspec-max-iters=0 -force-specialization -S < %s | FileCheck %s --check-prefix=DISABLED
   5
   6 ; DISABLED-NOT: @func.specialized.1(
   7 ; DISABLED-NOT: @func.specialized.2(
   8
   9 define internal i32 @func(ptr %0, i32 %1, ptr nocapture %2) {
  10   %4 = alloca i32, align 4
  11   store i32 %1, ptr %4, align 4
  12   %5 = load i32, ptr %4, align 4
  13   %6 = icmp slt i32 %5, 1
  14   br i1 %6, label %14, label %7
  15
  16 7:                                                ; preds = %3
  17   %8 = load i32, ptr %4, align 4
  18   %9 = sext i32 %8 to i64
  19   %10 = getelementptr inbounds i32, ptr %0, i64 %9
  20   call void %2(ptr %10)
  21   %11 = load i32, ptr %4, align 4
  22   %12 = add nsw i32 %11, -1
  23   %13 = call i32 @func(ptr %0, i32 %12, ptr %2)
  24   br label %14
  25
  26 14:                                               ; preds = %3, %7
  27   ret i32 0
  28 }
  29
  30 define internal void @increment(ptr nocapture %0) {
  31   %2 = load i32, ptr %0, align 4
  32   %3 = add nsw i32 %2, 1
  33   store i32 %3, ptr %0, align 4
  34   ret void
  35 }
  36
  37 define internal void @decrement(ptr nocapture %0) {
  38   %2 = load i32, ptr %0, align 4
  39   %3 = add nsw i32 %2, -1
  40   store i32 %3, ptr %0, align 4
  41   ret void
  42 }
  43
  44 define i32 @main(ptr %0, i32 %1) {
  45 ; CHECK:    call void @func.specialized.2(ptr [[TMP0:%.*]], i32 [[TMP1:%.*]])
  46   %3 = call i32 @func(ptr %0, i32 %1, ptr nonnull @increment)
  47 ; CHECK:    call void @func.specialized.1(ptr [[TMP0]], i32 0)
  48   %4 = call i32 @func(ptr %0, i32 %3, ptr nonnull @decrement)
  49 ; CHECK:    ret i32 0
  50   ret i32 %4
  51 }
  52
  53 ; CHECK: @func.specialized.1(
  54 ; CHECK:    [[TMP3:%.*]] = alloca i32, align 4
  55 ; CHECK:    store i32 [[TMP1:%.*]], ptr [[TMP3]], align 4
  56 ; CHECK:    [[TMP4:%.*]] = load i32, ptr [[TMP3]], align 4
  57 ; CHECK:    [[TMP5:%.*]] = icmp slt i32 [[TMP4]], 1
  58 ; CHECK:    br i1 [[TMP5]], label [[TMP13:%.*]], label [[TMP6:%.*]]
  59 ; CHECK:       6:
  60 ; CHECK:    [[TMP7:%.*]] = load i32, ptr [[TMP3]], align 4
  61 ; CHECK:    [[TMP8:%.*]] = sext i32 [[TMP7]] to i64
  62 ; CHECK:    [[TMP9:%.*]] = getelementptr inbounds i32, ptr [[TMP0:%.*]], i64 [[TMP8]]
  63 ; CHECK:    call void @decrement(ptr [[TMP9]])
  64 ; CHECK:    [[TMP10:%.*]] = load i32, ptr [[TMP3]], align 4
  65 ; CHECK:    [[TMP11:%.*]] = add nsw i32 [[TMP10]], -1
  66 ; CHECK:    call void @func.specialized.1(ptr [[TMP0]], i32 [[TMP11]])
  67 ; CHECK:    br label [[TMP12:%.*]]
  68 ; CHECK:       12:
  69 ; CHECK:    ret void
  70 ;
  71 ;
  72 ; CHECK: @func.specialized.2(
  73 ; CHECK:    [[TMP3:%.*]] = alloca i32, align 4
  74 ; CHECK:    store i32 [[TMP1:%.*]], ptr [[TMP3]], align 4
  75 ; CHECK:    [[TMP4:%.*]] = load i32, ptr [[TMP3]], align 4
  76 ; CHECK:    [[TMP5:%.*]] = icmp slt i32 [[TMP4]], 1
  77 ; CHECK:    br i1 [[TMP5]], label [[TMP13:%.*]], label [[TMP6:%.*]]
  78 ; CHECK:       6:
  79 ; CHECK:    [[TMP7:%.*]] = load i32, ptr [[TMP3]], align 4
  80 ; CHECK:    [[TMP8:%.*]] = sext i32 [[TMP7]] to i64
  81 ; CHECK:    [[TMP9:%.*]] = getelementptr inbounds i32, ptr [[TMP0:%.*]], i64 [[TMP8]]
  82 ; CHECK:    call void @increment(ptr [[TMP9]])
  83 ; CHECK:    [[TMP10:%.*]] = load i32, ptr [[TMP3]], align 4
  84 ; CHECK:    [[TMP11:%.*]] = add nsw i32 [[TMP10]], -1
  85 ; CHECK:    call void @func.specialized.2(ptr [[TMP0]], i32 [[TMP11]])
  86 ; CHECK:    br label [[TMP12:%.*]]
  87 ; CHECK:       12:
  88 ; CHECK:    ret void