[HLSL] Fix WaveBallot dxil op function name (#174901)
Just fixing a typo from a copy paste.
diff --git a/llvm/lib/Target/DirectX/DXIL.td b/llvm/lib/Target/DirectX/DXIL.td
index d84fa4a..6d04732 100644
--- a/llvm/lib/Target/DirectX/DXIL.td
+++ b/llvm/lib/Target/DirectX/DXIL.td
@@ -218,6 +218,7 @@
def waveAllOp : DXILOpClass;
def waveAllTrue : DXILOpClass;
def waveAnyTrue : DXILOpClass;
+ def waveBallot : DXILOpClass;
def waveGetLaneCount : DXILOpClass;
def waveGetLaneIndex : DXILOpClass;
def waveIsFirstLane : DXILOpClass;
@@ -1071,7 +1072,7 @@
let stages = [Stages<DXIL1_0, [all_stages]>];
}
-def WaveActiveBallot : DXILOp<118, waveAnyTrue> {
+def WaveActiveBallot : DXILOp<118, waveBallot> {
let Doc = "returns uint4 containing a bitmask of the evaluation of the boolean expression for all active lanes in the current wave.";
let intrinsics = [IntrinSelect<int_dx_wave_ballot>];
let arguments = [Int1Ty];
diff --git a/llvm/test/CodeGen/DirectX/WaveActiveBallot.ll b/llvm/test/CodeGen/DirectX/WaveActiveBallot.ll
index 5c3f8fc..cf6255d 100644
--- a/llvm/test/CodeGen/DirectX/WaveActiveBallot.ll
+++ b/llvm/test/CodeGen/DirectX/WaveActiveBallot.ll
@@ -2,7 +2,7 @@
define noundef <4 x i32> @wave_ballot_simple(i1 noundef %p1) {
entry:
-; CHECK: call <4 x i32> @dx.op.waveAnyTrue.void(i32 118, i1 %p1)
+; CHECK: call <4 x i32> @dx.op.waveBallot.void(i32 118, i1 %p1)
%ret = call <4 x i32> @llvm.dx.wave.ballot(i1 %p1)
ret <4 x i32> %ret
}