llvm/test/CodeGen/DirectX/BufferLoad-sm61.ll

   1 ; RUN: opt -S -dxil-op-lower %s | FileCheck %s
   2 ; Before SM6.2 ByteAddressBuffer and StructuredBuffer lower to bufferLoad.
   3
   4 target triple = "dxil-pc-shadermodel6.1-compute"
   5
   6 ; CHECK-LABEL: define void @loadf32_struct
   7 define void @loadf32_struct(i32 %index) {
   8   %buffer = call target("dx.RawBuffer", float, 0, 0, 0)
   9       @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_f32_0_0_0(
  10           i32 0, i32 0, i32 1, i32 0, i1 false)
  11
  12   ; CHECK: [[DATA:%.*]] = call %dx.types.ResRet.f32 @dx.op.bufferLoad.f32(i32 68, %dx.types.Handle %{{.*}}, i32 %index, i32 0)
  13   %load = call {float, i1}
  14       @llvm.dx.resource.load.rawbuffer.f32.tdx.RawBuffer_f32_0_0_0t(
  15           target("dx.RawBuffer", float, 0, 0, 0) %buffer,
  16           i32 %index,
  17           i32 0)
  18
  19   ret void
  20 }
  21
  22 ; CHECK-LABEL: define void @loadv4f32_byte
  23 define void @loadv4f32_byte(i32 %offset) {
  24   %buffer = call target("dx.RawBuffer", i8, 0, 0, 0)
  25       @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_i8_0_0_0(
  26           i32 0, i32 0, i32 1, i32 0, i1 false)
  27
  28   ; CHECK: [[DATA:%.*]] = call %dx.types.ResRet.f32 @dx.op.bufferLoad.f32(i32 68, %dx.types.Handle %{{.*}}, i32 %offset, i32 0)
  29   %load = call {<4 x float>, i1}
  30       @llvm.dx.resource.load.rawbuffer.f32.tdx.RawBuffer_i8_0_0_0t(
  31           target("dx.RawBuffer", i8, 0, 0, 0) %buffer,
  32           i32 %offset,
  33           i32 0)
  34
  35   ret void
  36 }
  37
  38 ; CHECK-LABEL: define void @loadnested
  39 define void @loadnested(i32 %index) {
  40   %buffer = call
  41       target("dx.RawBuffer", {i32, {<4 x float>, <3 x half>}}, 0, 0, 0)
  42       @llvm.dx.resource.handlefrombinding(i32 0, i32 0, i32 1, i32 0, i1 false)
  43
  44   ; CHECK: [[DATAI32:%.*]] = call %dx.types.ResRet.i32 @dx.op.bufferLoad.i32(i32 68, %dx.types.Handle %{{.*}}, i32 %index, i32 0)
  45   %loadi32 = call {i32, i1} @llvm.dx.resource.load.rawbuffer.i32(
  46       target("dx.RawBuffer", {i32, {<4 x float>, <3 x half>}}, 0, 0, 0) %buffer,
  47       i32 %index, i32 0)
  48
  49   ; CHECK: [[DATAF32:%.*]] = call %dx.types.ResRet.f32 @dx.op.bufferLoad.f32(i32 68, %dx.types.Handle %{{.*}}, i32 %index, i32 4)
  50   %loadf32 = call {<4 x float>, i1} @llvm.dx.resource.load.rawbuffer.v4f32(
  51       target("dx.RawBuffer", {i32, {<4 x float>, <3 x half>}}, 0, 0, 0) %buffer,
  52       i32 %index, i32 4)
  53
  54   ; CHECK: [[DATAF16:%.*]] = call %dx.types.ResRet.f16 @dx.op.bufferLoad.f16(i32 68, %dx.types.Handle %{{.*}}, i32 %index, i32 20)
  55   %loadf16 = call {<3 x half>, i1} @llvm.dx.resource.load.rawbuffer.v3f16(
  56       target("dx.RawBuffer", {i32, {<4 x float>, <3 x half>}}, 0, 0, 0) %buffer,
  57       i32 %index, i32 20)
  58
  59   ret void
  60 }