test/CodeGen/AMDGPU/si-scheduler.ll

   1 ; FIXME: The si scheduler crashes if when lane mask tracking is enabled, so
   2 ; we need to disable this when the si scheduler is being used.
   3 ; The only way the subtarget knows that the si machine scheduler is being used
   4 ; is to specify -mattr=si-scheduler.  If we just pass --misched=si, the backend
   5 ; won't know what scheduler we are using.
   6 ; RUN: llc -march=amdgcn --misched=si -mattr=si-scheduler < %s | FileCheck %s
   7 ; RUN: llc -march=amdgcn -mcpu=gfx1010 --misched=si -mattr=si-scheduler < %s | FileCheck %s
   8
   9 ; The test checks the "si" machine scheduler pass works correctly.
  10
  11 ; CHECK-LABEL: {{^}}main:
  12 ; CHECK: s_wqm
  13 ; CHECK: s_load_dwordx8
  14 ; CHECK: s_load_dwordx4
  15 ; CHECK: s_waitcnt lgkmcnt(0)
  16 ; CHECK: image_sample
  17 ; CHECK: s_waitcnt vmcnt(0)
  18 ; CHECK: exp
  19 ; CHECK: s_endpgm
  20 define amdgpu_ps void @main([6 x <16 x i8>] addrspace(4)* inreg %arg, [17 x <16 x i8>] addrspace(4)* inreg %arg1, [17 x <4 x i32>] addrspace(4)* inreg %arg2, [34 x <8 x i32>] addrspace(4)* inreg %arg3, float inreg %arg4, i32 inreg %arg5, <2 x i32> %arg6, <2 x i32> %arg7, <2 x i32> %arg8, <3 x i32> %arg9, <2 x i32> %arg10, <2 x i32> %arg11, <2 x i32> %arg12, float %arg13, float %arg14, float %arg15, float %arg16, float %arg17, float %arg18, i32 %arg19, float %arg20, float %arg21) #0 {
  21 main_body:
  22   %tmp = bitcast [34 x <8 x i32>] addrspace(4)* %arg3 to <32 x i8> addrspace(4)*
  23   %tmp22 = load <32 x i8>, <32 x i8> addrspace(4)* %tmp, align 32, !tbaa !0
  24   %tmp23 = bitcast [17 x <4 x i32>] addrspace(4)* %arg2 to <16 x i8> addrspace(4)*
  25   %tmp24 = load <16 x i8>, <16 x i8> addrspace(4)* %tmp23, align 16, !tbaa !0
  26   %i.i = extractelement <2 x i32> %arg11, i32 0
  27   %j.i = extractelement <2 x i32> %arg11, i32 1
  28   %i.f.i = bitcast i32 %i.i to float
  29   %j.f.i = bitcast i32 %j.i to float
  30   %p1.i = call float @llvm.amdgcn.interp.p1(float %i.f.i, i32 0, i32 0, i32 %arg5) #1
  31   %p2.i = call float @llvm.amdgcn.interp.p2(float %p1.i, float %j.f.i, i32 0, i32 0, i32 %arg5) #1
  32   %i.i1 = extractelement <2 x i32> %arg11, i32 0
  33   %j.i2 = extractelement <2 x i32> %arg11, i32 1
  34   %i.f.i3 = bitcast i32 %i.i1 to float
  35   %j.f.i4 = bitcast i32 %j.i2 to float
  36   %p1.i5 = call float @llvm.amdgcn.interp.p1(float %i.f.i3, i32 1, i32 0, i32 %arg5) #1
  37   %p2.i6 = call float @llvm.amdgcn.interp.p2(float %p1.i5, float %j.f.i4, i32 1, i32 0, i32 %arg5) #1
  38   %tmp22.bc = bitcast <32 x i8> %tmp22 to <8 x i32>
  39   %tmp24.bc = bitcast <16 x i8> %tmp24 to <4 x i32>
  40   %tmp31 = call <4 x float> @llvm.amdgcn.image.sample.2d.v4f32.f32(i32 15, float %p2.i, float %p2.i6, <8 x i32> %tmp22.bc, <4 x i32> %tmp24.bc, i1 0, i32 0, i32 0)
  41
  42   %tmp32 = extractelement <4 x float> %tmp31, i32 0
  43   %tmp33 = extractelement <4 x float> %tmp31, i32 1
  44   %tmp34 = extractelement <4 x float> %tmp31, i32 2
  45   %tmp35 = extractelement <4 x float> %tmp31, i32 3
  46   %tmp36 = call <2 x half> @llvm.amdgcn.cvt.pkrtz(float %tmp32, float %tmp33)
  47   %tmp38 = call <2 x half> @llvm.amdgcn.cvt.pkrtz(float %tmp34, float %tmp35)
  48   call void @llvm.amdgcn.exp.compr.v2f16(i32 0, i32 15, <2 x half> %tmp36, <2 x half> %tmp38, i1 true, i1 false) #0
  49   ret void
  50 }
  51
  52 declare float @llvm.amdgcn.interp.p1(float, i32, i32, i32) #1
  53 declare float @llvm.amdgcn.interp.p2(float, float, i32, i32, i32) #1
  54 declare void @llvm.amdgcn.exp.compr.v2f16(i32, i32, <2 x half>, <2 x half>, i1, i1) #0
  55 declare <2 x half> @llvm.amdgcn.cvt.pkrtz(float, float) #1
  56 declare <4 x float> @llvm.amdgcn.image.sample.2d.v4f32.f32(i32, float, float, <8 x i32>, <4 x i32>, i1, i32, i32) #2
  57
  58 attributes #0 = { nounwind }
  59 attributes #1 = { nounwind readnone }
  60 attributes #2 = { nounwind readonly }
  61
  62 !0 = !{!1, !1, i64 0, i32 1}
  63 !1 = !{!"const", !2}
  64 !2 = !{!"tbaa root"}
  65
  66
  67 ; CHECK-LABEL: amdgpu_ps_main:
  68 ; CHECK s_buffer_load_dword
  69 define amdgpu_ps void @_amdgpu_ps_main(i32 %arg) local_unnamed_addr {
  70 .entry:
  71   %tmp = insertelement <2 x i32> zeroinitializer, i32 %arg, i32 0
  72   %tmp1 = bitcast <2 x i32> %tmp to i64
  73   %tmp2 = inttoptr i64 %tmp1 to <4 x i32> addrspace(4)*
  74   %tmp3 = load <4 x i32>, <4 x i32> addrspace(4)* %tmp2, align 16
  75   %tmp4 = tail call i32 @llvm.amdgcn.s.buffer.load.i32(<4 x i32> %tmp3, i32 0, i32 0) #0
  76   switch i32 %tmp4, label %bb [
  77     i32 0, label %bb5
  78     i32 1, label %bb6
  79   ]
  80
  81 bb:                                               ; preds = %.entry
  82   unreachable
  83
  84 bb5:                                              ; preds = %.entry
  85   unreachable
  86
  87 bb6:                                              ; preds = %.entry
  88   unreachable
  89 }
  90
  91 ; Function Attrs: nounwind readnone
  92 declare i32 @llvm.amdgcn.s.buffer.load.i32(<4 x i32>, i32, i32 immarg) #1