llvm/test/Transforms/OpenMP/fold_generic_main_thread.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --function-signature
   2 ; RUN: opt -S -passes='openmp-opt' < %s | FileCheck %s
   3 ; RUN: opt -openmp-opt-disable-folding -S -passes='openmp-opt' < %s | FileCheck %s --check-prefix=CHECK-DISABLED
   4 ; ModuleID = 'single_threaded_exeuction.c'
   5
   6 %struct.ident_t = type { i32, i32, i32, i32, i8* }
   7
   8 @0 = private unnamed_addr constant [1 x i8] c"\00", align 1
   9 @1 = private unnamed_addr constant %struct.ident_t { i32 0, i32 2, i32 0, i32 0, i8* getelementptr inbounds ([1 x i8], [1 x i8]* @0, i32 0, i32 0) }, align 8
  10
  11 define void @kernel() {
  12 ; CHECK-LABEL: define {{[^@]+}}@kernel() {
  13 ; CHECK-NEXT:    [[CALL:%.*]] = call i32 @__kmpc_target_init(%struct.ident_t* nonnull @[[GLOB1:[0-9]+]], i1 false, i1 false, i1 false)
  14 ; CHECK-NEXT:    [[CMP:%.*]] = icmp eq i32 [[CALL]], -1
  15 ; CHECK-NEXT:    br i1 [[CMP]], label [[IF_THEN:%.*]], label [[IF_ELSE:%.*]]
  16 ; CHECK:       if.then:
  17 ; CHECK-NEXT:    call void @foo()
  18 ; CHECK-NEXT:    br label [[IF_END:%.*]]
  19 ; CHECK:       if.else:
  20 ; CHECK-NEXT:    call void @bar()
  21 ; CHECK-NEXT:    br label [[IF_END]]
  22 ; CHECK:       if.end:
  23 ; CHECK-NEXT:    call void @__kmpc_target_deinit(%struct.ident_t* null, i1 false, i1 true)
  24 ; CHECK-NEXT:    ret void
  25 ;
  26 ; CHECK-DISABLED-LABEL: define {{[^@]+}}@kernel() {
  27 ; CHECK-DISABLED-NEXT:    [[CALL:%.*]] = call i32 @__kmpc_target_init(%struct.ident_t* nonnull @[[GLOB1:[0-9]+]], i1 false, i1 false, i1 false)
  28 ; CHECK-DISABLED-NEXT:    [[CMP:%.*]] = icmp eq i32 [[CALL]], -1
  29 ; CHECK-DISABLED-NEXT:    br i1 [[CMP]], label [[IF_THEN:%.*]], label [[IF_ELSE:%.*]]
  30 ; CHECK-DISABLED:       if.then:
  31 ; CHECK-DISABLED-NEXT:    call void @foo()
  32 ; CHECK-DISABLED-NEXT:    br label [[IF_END:%.*]]
  33 ; CHECK-DISABLED:       if.else:
  34 ; CHECK-DISABLED-NEXT:    call void @bar()
  35 ; CHECK-DISABLED-NEXT:    br label [[IF_END]]
  36 ; CHECK-DISABLED:       if.end:
  37 ; CHECK-DISABLED-NEXT:    call void @__kmpc_target_deinit(%struct.ident_t* null, i1 false, i1 true)
  38 ; CHECK-DISABLED-NEXT:    ret void
  39 ;
  40   %call = call i32 @__kmpc_target_init(%struct.ident_t* nonnull @1, i1 false, i1 false, i1 false)
  41   %cmp = icmp eq i32 %call, -1
  42   br i1 %cmp, label %if.then, label %if.else
  43 if.then:
  44   call void @foo()
  45   br label %if.end
  46 if.else:
  47   call void @bar()
  48   br label %if.end
  49 if.end:
  50   call void @__kmpc_target_deinit(%struct.ident_t* null, i1 false, i1 true)
  51   ret void
  52 }
  53
  54 ; Function Attrs: noinline
  55 define internal void @foo() {
  56 ; CHECK-LABEL: define {{[^@]+}}@foo() {
  57 ; CHECK-NEXT:  entry:
  58 ; CHECK-NEXT:    [[TID:%.*]] = call i32 @__kmpc_get_hardware_thread_id()
  59 ; CHECK-NEXT:    br label [[IF_THEN:%.*]]
  60 ; CHECK:       if.then:
  61 ; CHECK-NEXT:    call void @baz()
  62 ; CHECK-NEXT:    br label [[IF_END:%.*]]
  63 ; CHECK:       if.end:
  64 ; CHECK-NEXT:    ret void
  65 ;
  66 ; CHECK-DISABLED-LABEL: define {{[^@]+}}@foo() {
  67 ; CHECK-DISABLED-NEXT:  entry:
  68 ; CHECK-DISABLED-NEXT:    [[TID:%.*]] = call i32 @__kmpc_get_hardware_thread_id()
  69 ; CHECK-DISABLED-NEXT:    [[ISMAIN:%.*]] = call signext i8 @__kmpc_is_generic_main_thread_id(i32 [[TID]])
  70 ; CHECK-DISABLED-NEXT:    [[PRED:%.*]] = icmp eq i8 [[ISMAIN]], 1
  71 ; CHECK-DISABLED-NEXT:    br i1 [[PRED]], label [[IF_THEN:%.*]], label [[IF_END:%.*]]
  72 ; CHECK-DISABLED:       if.then:
  73 ; CHECK-DISABLED-NEXT:    call void @baz()
  74 ; CHECK-DISABLED-NEXT:    br label [[IF_END]]
  75 ; CHECK-DISABLED:       if.end:
  76 ; CHECK-DISABLED-NEXT:    ret void
  77 ;
  78 entry:
  79   %tid = call i32 @__kmpc_get_hardware_thread_id();
  80   %ismain = call signext i8 @__kmpc_is_generic_main_thread_id(i32 %tid)
  81   %pred = icmp eq i8 %ismain, 1
  82   br i1 %pred, label %if.then, label %if.end
  83
  84 if.then:
  85   call void @baz()
  86   br label %if.end
  87
  88 if.end:
  89   ret void
  90 }
  91
  92 ; Function Attrs: noinline
  93 define internal void @bar() {
  94 ; CHECK-LABEL: define {{[^@]+}}@bar() {
  95 ; CHECK-NEXT:  entry:
  96 ; CHECK-NEXT:    [[TID:%.*]] = call i32 @__kmpc_get_hardware_thread_id()
  97 ; CHECK-NEXT:    [[ISMAIN:%.*]] = call signext i8 @__kmpc_is_generic_main_thread_id(i32 [[TID]])
  98 ; CHECK-NEXT:    [[PRED:%.*]] = icmp eq i8 [[ISMAIN]], 1
  99 ; CHECK-NEXT:    br i1 [[PRED]], label [[IF_THEN:%.*]], label [[IF_END:%.*]]
 100 ; CHECK:       if.then:
 101 ; CHECK-NEXT:    call void @baz()
 102 ; CHECK-NEXT:    br label [[IF_END]]
 103 ; CHECK:       if.end:
 104 ; CHECK-NEXT:    ret void
 105 ;
 106 ; CHECK-DISABLED-LABEL: define {{[^@]+}}@bar() {
 107 ; CHECK-DISABLED-NEXT:  entry:
 108 ; CHECK-DISABLED-NEXT:    [[TID:%.*]] = call i32 @__kmpc_get_hardware_thread_id()
 109 ; CHECK-DISABLED-NEXT:    [[ISMAIN:%.*]] = call signext i8 @__kmpc_is_generic_main_thread_id(i32 [[TID]])
 110 ; CHECK-DISABLED-NEXT:    [[PRED:%.*]] = icmp eq i8 [[ISMAIN]], 1
 111 ; CHECK-DISABLED-NEXT:    br i1 [[PRED]], label [[IF_THEN:%.*]], label [[IF_END:%.*]]
 112 ; CHECK-DISABLED:       if.then:
 113 ; CHECK-DISABLED-NEXT:    call void @baz()
 114 ; CHECK-DISABLED-NEXT:    br label [[IF_END]]
 115 ; CHECK-DISABLED:       if.end:
 116 ; CHECK-DISABLED-NEXT:    ret void
 117 ;
 118 entry:
 119   %tid = call i32 @__kmpc_get_hardware_thread_id();
 120   %ismain = call signext i8 @__kmpc_is_generic_main_thread_id(i32 %tid)
 121   %pred = icmp eq i8 %ismain, 1
 122   br i1 %pred, label %if.then, label %if.end
 123
 124 if.then:
 125   call void @baz()
 126   br label %if.end
 127
 128 if.end:
 129   ret void
 130 }
 131
 132 declare void @baz()
 133
 134 declare i8 @__kmpc_is_generic_main_thread_id(i32)
 135
 136 declare i32 @__kmpc_get_hardware_thread_id()
 137
 138 declare i32 @__kmpc_target_init(%struct.ident_t*, i1, i1, i1)
 139
 140 declare void @__kmpc_target_deinit(%struct.ident_t*, i1, i1)
 141
 142 !llvm.dbg.cu = !{!0}
 143 !llvm.module.flags = !{!3, !4, !5, !6}
 144 !nvvm.annotations = !{!7}
 145
 146 !0 = distinct !DICompileUnit(language: DW_LANG_C99, file: !1, producer: "clang version 13.0.0", isOptimized: false, runtimeVersion: 0, emissionKind: FullDebug, enums: !2, splitDebugInlining: false, nameTableKind: None)
 147 !1 = !DIFile(filename: "fold_generic_main_thread.c", directory: "/tmp/fold_generic_main_thread.c")
 148 !2 = !{}
 149 !3 = !{i32 2, !"Debug Info Version", i32 3}
 150 !4 = !{i32 1, !"wchar_size", i32 4}
 151 !5 = !{i32 7, !"openmp", i32 50}
 152 !6 = !{i32 7, !"openmp-device", i32 50}
 153 !7 = !{void ()* @kernel, !"kernel", i32 1}