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