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

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --function-signature --check-globals
   2 ; RUN: opt -S -mtriple=amdgcn-amd- -passes=amdgpu-attributor %s | FileCheck %s
   3
   4 @x = global i32 0
   5
   6 ; Propagate the uniform-work-group-attribute from the kernel to callee if it doesn't have it
   7 ;.
   8 ; CHECK: @[[X:[a-zA-Z0-9_$"\\.-]+]] = global i32 0
   9 ;.
  10 define void @func() #0 {
  11 ; CHECK-LABEL: define {{[^@]+}}@func
  12 ; CHECK-SAME: () #[[ATTR0:[0-9]+]] {
  13 ; CHECK-NEXT:    store i32 0, ptr @x, align 4
  14 ; CHECK-NEXT:    ret void
  15 ;
  16   store i32 0, ptr @x
  17   ret void
  18 }
  19
  20 define amdgpu_kernel void @kernel1() #1 {
  21 ; CHECK-LABEL: define {{[^@]+}}@kernel1
  22 ; CHECK-SAME: () #[[ATTR1:[0-9]+]] {
  23 ; CHECK-NEXT:    call void @func()
  24 ; CHECK-NEXT:    ret void
  25 ;
  26   call void @func()
  27   ret void
  28 }
  29
  30 ; External declaration of a function
  31 define weak_odr void @weak_func() #0 {
  32 ; CHECK-LABEL: define {{[^@]+}}@weak_func
  33 ; CHECK-SAME: () #[[ATTR2:[0-9]+]] {
  34 ; CHECK-NEXT:    store i32 0, ptr @x, align 4
  35 ; CHECK-NEXT:    ret void
  36 ;
  37   store i32 0, ptr @x
  38   ret void
  39 }
  40
  41 define amdgpu_kernel void @kernel2() #2 {
  42 ; CHECK-LABEL: define {{[^@]+}}@kernel2
  43 ; CHECK-SAME: () #[[ATTR3:[0-9]+]] {
  44 ; CHECK-NEXT:    call void @weak_func()
  45 ; CHECK-NEXT:    ret void
  46 ;
  47   call void @weak_func()
  48   ret void
  49 }
  50
  51 attributes #0 = { nounwind }
  52 attributes #1 = { "uniform-work-group-size"="false" }
  53 attributes #2 = { "uniform-work-group-size"="true" }
  54 ;.
  55 ; CHECK: attributes #[[ATTR0]] = { nounwind "amdgpu-no-agpr" "amdgpu-no-completion-action" "amdgpu-no-default-queue" "amdgpu-no-dispatch-id" "amdgpu-no-dispatch-ptr" "amdgpu-no-heap-ptr" "amdgpu-no-hostcall-ptr" "amdgpu-no-implicitarg-ptr" "amdgpu-no-lds-kernel-id" "amdgpu-no-multigrid-sync-arg" "amdgpu-no-queue-ptr" "amdgpu-no-workgroup-id-x" "amdgpu-no-workgroup-id-y" "amdgpu-no-workgroup-id-z" "amdgpu-no-workitem-id-x" "amdgpu-no-workitem-id-y" "amdgpu-no-workitem-id-z" "amdgpu-waves-per-eu"="4,10" "uniform-work-group-size"="false" }
  56 ; CHECK: attributes #[[ATTR1]] = { "amdgpu-no-agpr" "amdgpu-no-completion-action" "amdgpu-no-default-queue" "amdgpu-no-dispatch-id" "amdgpu-no-dispatch-ptr" "amdgpu-no-heap-ptr" "amdgpu-no-hostcall-ptr" "amdgpu-no-implicitarg-ptr" "amdgpu-no-lds-kernel-id" "amdgpu-no-multigrid-sync-arg" "amdgpu-no-queue-ptr" "amdgpu-no-workgroup-id-x" "amdgpu-no-workgroup-id-y" "amdgpu-no-workgroup-id-z" "amdgpu-no-workitem-id-x" "amdgpu-no-workitem-id-y" "amdgpu-no-workitem-id-z" "uniform-work-group-size"="false" }
  57 ; CHECK: attributes #[[ATTR2]] = { nounwind "amdgpu-waves-per-eu"="4,10" "uniform-work-group-size"="false" }
  58 ; CHECK: attributes #[[ATTR3]] = { "uniform-work-group-size"="true" }
  59 ;.