test/Transforms/LoopUnroll/AMDGPU/unroll-cost-addrspacecast.ll

   1 ; RUN: opt -S -mtriple=amdgcn-unknown-amdhsa -mcpu=hawaii -loop-unroll -unroll-threshold=75 -unroll-peel-count=0 -unroll-allow-partial=false -unroll-max-iteration-count-to-analyze=16 < %s | FileCheck %s
   2
   3 ; CHECK-LABEL: @test_func_addrspacecast_cost_noop(
   4 ; CHECK-NOT: br i1
   5 define amdgpu_kernel void @test_func_addrspacecast_cost_noop(float addrspace(1)* noalias nocapture %out, float addrspace(1)* noalias nocapture %in) #0 {
   6 entry:
   7   br label %for.body
   8
   9 for.body:
  10   %indvars.iv = phi i32 [ %indvars.iv.next, %for.body ], [ 0, %entry ]
  11   %sum.02 = phi float [ %fmul, %for.body ], [ 0.0, %entry ]
  12   %arrayidx.in = getelementptr inbounds float, float addrspace(1)* %in, i32 %indvars.iv
  13   %arrayidx.out = getelementptr inbounds float, float addrspace(1)* %out, i32 %indvars.iv
  14   %cast.in = addrspacecast float addrspace(1)* %arrayidx.in to float*
  15   %cast.out = addrspacecast float addrspace(1)* %arrayidx.out to float*
  16   %load = load float, float* %cast.in
  17   %fmul = fmul float %load, %sum.02
  18   store float %fmul, float* %cast.out
  19   %indvars.iv.next = add i32 %indvars.iv, 1
  20   %exitcond = icmp eq i32 %indvars.iv.next, 16
  21   br i1 %exitcond, label %for.end, label %for.body
  22
  23 for.end:
  24   ret void
  25 }
  26
  27 ; Free, but not a no-op
  28 ; CHECK-LABEL: @test_func_addrspacecast_cost_free(
  29 ; CHECK-NOT: br i1
  30 define amdgpu_kernel void @test_func_addrspacecast_cost_free(float* noalias nocapture %out, float* noalias nocapture %in) #0 {
  31 entry:
  32   br label %for.body
  33
  34 for.body:
  35   %indvars.iv = phi i32 [ %indvars.iv.next, %for.body ], [ 0, %entry ]
  36   %sum.02 = phi float [ %fmul, %for.body ], [ 0.0, %entry ]
  37   %arrayidx.in = getelementptr inbounds float, float* %in, i32 %indvars.iv
  38   %arrayidx.out = getelementptr inbounds float, float* %out, i32 %indvars.iv
  39   %cast.in = addrspacecast float* %arrayidx.in to float addrspace(3)*
  40   %cast.out = addrspacecast float* %arrayidx.out to float addrspace(3)*
  41   %load = load float, float addrspace(3)* %cast.in
  42   %fmul = fmul float %load, %sum.02
  43   store float %fmul, float addrspace(3)* %cast.out
  44   %indvars.iv.next = add i32 %indvars.iv, 1
  45   %exitcond = icmp eq i32 %indvars.iv.next, 16
  46   br i1 %exitcond, label %for.end, label %for.body
  47
  48 for.end:
  49   ret void
  50 }
  51
  52 ; CHECK-LABEL: @test_func_addrspacecast_cost_nonfree(
  53 ; CHECK: br i1 %exitcond
  54 define amdgpu_kernel void @test_func_addrspacecast_cost_nonfree(float addrspace(3)* noalias nocapture %out, float addrspace(3)* noalias nocapture %in) #0 {
  55 entry:
  56   br label %for.body
  57
  58 for.body:
  59   %indvars.iv = phi i32 [ %indvars.iv.next, %for.body ], [ 0, %entry ]
  60   %sum.02 = phi float [ %fmul, %for.body ], [ 0.0, %entry ]
  61   %arrayidx.in = getelementptr inbounds float, float addrspace(3)* %in, i32 %indvars.iv
  62   %arrayidx.out = getelementptr inbounds float, float addrspace(3)* %out, i32 %indvars.iv
  63   %cast.in = addrspacecast float addrspace(3)* %arrayidx.in to float*
  64   %cast.out = addrspacecast float addrspace(3)* %arrayidx.out to float*
  65   %load = load float, float* %cast.in
  66   %fmul = fmul float %load, %sum.02
  67   store float %fmul, float* %cast.out
  68   %indvars.iv.next = add i32 %indvars.iv, 1
  69   %exitcond = icmp eq i32 %indvars.iv.next, 16
  70   br i1 %exitcond, label %for.end, label %for.body
  71
  72 for.end:
  73   ret void
  74 }
  75
  76 attributes #0 = { nounwind }
  77 attributes #1 = { nounwind readnone speculatable }