llvm/test/CodeGen/DirectX/discard.ll

   1 ; RUN: opt -passes='function(scalarizer),module(dxil-op-lower,dxil-intrinsic-expansion)' -S -mtriple=dxil-pc-shadermodel6.3-pixel %s | FileCheck %s
   2
   3 ; CHECK-LABEL: define void @test_scalar
   4 ; CHECK: call void @dx.op.discard(i32 82, i1 %0)
   5 ;
   6 define void @test_scalar(float noundef %p) #0 {
   7 entry:
   8   %0 = fcmp olt float %p, 0.000000e+00
   9   call void @llvm.dx.discard(i1 %0)
  10   ret void
  11 }
  12
  13 ; CHECK-LABEL: define void @test_vector
  14 ; CHECK:       [[EXTR0:%.*]] = extractelement <4 x i1> [[INP:%.*]], i64 0
  15 ; CHECK-NEXT:  [[EXTR1:%.*]] = extractelement <4 x i1> [[INP:%.*]], i64 1
  16 ; CHECK-NEXT:  [[OR1:%.*]] = or i1 [[EXTR0]], [[EXTR1]]
  17 ; CHECK-NEXT:  [[EXTR2:%.*]] = extractelement <4 x i1> [[INP:%.*]], i64 2
  18 ; CHECK-NEXT:  [[OR2:%.*]] = or i1 [[OR1]], [[EXTR2]]
  19 ; CHECK-NEXT:  [[EXTR3:%.*]] = extractelement <4 x i1> [[INP:%.*]], i64 3
  20 ; CHECK-NEXT:  [[OR3:%.*]] = or i1 [[OR2]], [[EXTR3]]
  21 ; CHECK-NEXT:  call void @dx.op.discard(i32 82, i1 [[OR3]])
  22 ;
  23 define void @test_vector(<4 x float> noundef %p) #0 {
  24 entry:
  25   %0 = fcmp olt <4 x float> %p, zeroinitializer
  26   %1 = call i1 @llvm.dx.any.v4i1(<4 x i1> %0)
  27   call void @llvm.dx.discard(i1 %1)
  28   ret void
  29 }
  30
  31 declare void @llvm.dx.discard(i1)