llvm/test/CodeGen/AMDGPU/uniform-work-group-prevent-attribute-propagation.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --function-signature --check-globals
   2
   3 ; RUN: opt -S -mtriple=amdgcn-amd- -amdgpu-annotate-kernel-features < %s | FileCheck -check-prefixes=CHECK,AKF_CHECK %s
   4 ; RUN: opt -S -mtriple=amdgcn-amd- -amdgpu-attributor < %s | FileCheck -check-prefixes=CHECK,ATTRIBUTOR_CHECK %s
   5
   6 ; Function added to prevent attributor from deleting call sites.
   7
   8 ; Two kernels with different values of the uniform-work-group-attribute call the same function
   9 @x = global i32 0
  10
  11 ;.
  12 ; CHECK: @[[X:[a-zA-Z0-9_$"\\.-]+]] = global i32 0
  13 ;.
  14 define void @func() #0 {
  15 ; CHECK-LABEL: define {{[^@]+}}@func
  16 ; CHECK-SAME: () #[[ATTR0:[0-9]+]] {
  17 ; CHECK-NEXT:    store i32 0, i32* @x, align 4
  18 ; CHECK-NEXT:    ret void
  19 ;
  20   store i32 0, i32* @x
  21   ret void
  22 }
  23
  24 define amdgpu_kernel void @kernel1() #1 {
  25 ; AKF_CHECK-LABEL: define {{[^@]+}}@kernel1
  26 ; AKF_CHECK-SAME: () #[[ATTR1:[0-9]+]] {
  27 ; AKF_CHECK-NEXT:    call void @func()
  28 ; AKF_CHECK-NEXT:    ret void
  29 ;
  30 ; ATTRIBUTOR_CHECK-LABEL: define {{[^@]+}}@kernel1
  31 ; ATTRIBUTOR_CHECK-SAME: () #[[ATTR1:[0-9]+]] {
  32 ; ATTRIBUTOR_CHECK-NEXT:    call void @func() #[[ATTR3:[0-9]+]]
  33 ; ATTRIBUTOR_CHECK-NEXT:    ret void
  34 ;
  35   call void @func()
  36   ret void
  37 }
  38
  39 define amdgpu_kernel void @kernel2() #2 {
  40 ; AKF_CHECK-LABEL: define {{[^@]+}}@kernel2
  41 ; AKF_CHECK-SAME: () #[[ATTR2:[0-9]+]] {
  42 ; AKF_CHECK-NEXT:    call void @func()
  43 ; AKF_CHECK-NEXT:    ret void
  44 ;
  45 ; ATTRIBUTOR_CHECK-LABEL: define {{[^@]+}}@kernel2
  46 ; ATTRIBUTOR_CHECK-SAME: () #[[ATTR2:[0-9]+]] {
  47 ; ATTRIBUTOR_CHECK-NEXT:    call void @func() #[[ATTR3]]
  48 ; ATTRIBUTOR_CHECK-NEXT:    ret void
  49 ;
  50   call void @func()
  51   ret void
  52 }
  53
  54 attributes #1 = { "uniform-work-group-size"="true" }
  55
  56 ;.
  57 ; AKF_CHECK: attributes #[[ATTR0]] = { "uniform-work-group-size"="false" }
  58 ; AKF_CHECK: attributes #[[ATTR1]] = { "amdgpu-calls" "uniform-work-group-size"="true" }
  59 ; AKF_CHECK: attributes #[[ATTR2]] = { "amdgpu-calls" "uniform-work-group-size"="false" }
  60 ;.
  61 ; ATTRIBUTOR_CHECK: attributes #[[ATTR0]] = { nounwind writeonly "uniform-work-group-size"="false" }
  62 ; ATTRIBUTOR_CHECK: attributes #[[ATTR1]] = { "amdgpu-calls" "uniform-work-group-size"="true" }
  63 ; ATTRIBUTOR_CHECK: attributes #[[ATTR2]] = { "amdgpu-calls" "uniform-work-group-size"="false" }
  64 ; ATTRIBUTOR_CHECK: attributes #[[ATTR3]] = { nounwind writeonly }
  65 ;.