test/CodeGen/AMDGPU/uniform-work-group-recursion-test.ll

   1 ; RUN: opt -S -mtriple=amdgcn-amd- -amdgpu-annotate-kernel-features %s | FileCheck %s
   2
   3 ; Test to ensure recursive functions exhibit proper behaviour
   4 ; Test to generate fibonacci numbers
   5
   6 ; CHECK: define i32 @fib(i32 %n) #[[FIB:[0-9]+]] {
   7 define i32 @fib(i32 %n) #0 {
   8   %cmp1 = icmp eq i32 %n, 0
   9   br i1 %cmp1, label %exit, label %cont1
  10
  11 cont1:
  12   %cmp2 = icmp eq i32 %n, 1
  13   br i1 %cmp2, label %exit, label %cont2
  14
  15 cont2:
  16   %nm1 = sub i32 %n, 1
  17   %fibm1 = call i32 @fib(i32 %nm1)
  18   %nm2 = sub i32 %n, 2
  19   %fibm2 = call i32 @fib(i32 %nm2)
  20   %retval = add i32 %fibm1, %fibm2
  21
  22   ret i32 %retval
  23
  24 exit:
  25   ret i32 1
  26 }
  27
  28 ; CHECK: define amdgpu_kernel void @kernel(i32 addrspace(1)* %m) #[[FIB]] {
  29 define amdgpu_kernel void @kernel(i32 addrspace(1)* %m) #1 {
  30   %r = call i32 @fib(i32 5)
  31   store i32 %r, i32 addrspace(1)* %m
  32   ret void
  33 }
  34
  35 attributes #1 = { "uniform-work-group-size"="true" }
  36
  37 ; CHECK: attributes #[[FIB]] = { "uniform-work-group-size"="true" }