Add WaveActiveBallot tests (#645)

Fixes #96

Author: bob80905

test/WaveOps/WaveActiveBallot.Wave128.test

@@ -0,0 +1,62 @@
+#--- source.hlsl
+RWStructuredBuffer<uint4> Out : register(u1);
+
+[WaveSize(128)]
+[numthreads(128, 1, 1)]
+void main(uint3 threadID : SV_DispatchThreadID) {
+  // We expect all resulting uint's bitmask to be 
+  // 0xBFFFFFFF
+  Out[0] = WaveActiveBallot(threadID.x != 30 && threadID.x != 62 && threadID.x != 94 && threadID.x != 126);
+
+  // We expect the first resulting uint's bitmask to be 
+  // 0x0000FFFF, and the second to be 0xFFFF0000
+  // third and fourth should be 0xFFFFFF00 and 0x00FFFFFF
+  Out[1] = WaveActiveBallot(threadID.x < 16 || (threadID.x >= 48 && threadID.x < 64) ||
+  (threadID.x >= 72 && threadID.x < 96) || (threadID.x >= 96 && threadID.x < 120));
+
+  // We expect the resulting uint4 to be 4 0x00000000's
+  Out[2] = WaveActiveBallot(false);
+
+  // We expect all resulting uint's bitmask to be 
+  // 0xAAAAAAAA
+  Out[3] = WaveActiveBallot(threadID.x % 2 == 1);
+}
+
+//--- pipeline.yaml
+
+---
+Shaders:
+  - Stage: Compute
+    Entry: main
+    DispatchSize: [1, 1, 1]
+Buffers:  
+  - Name: Out
+    Format: UInt32
+    Stride: 4
+    FillSize: 64
+  - Name: ExpectedOut
+    Format: UInt32
+    Stride: 4
+    Data: [3221225471, 3221225471, 3221225471, 3221225471, 65535, 4294901760, 4294967040, 16777215, 0, 0, 0, 0, 2863311530, 2863311530, 2863311530, 2863311530]
+Results:
+  - Result: Test
+    Rule: BufferExact
+    Actual: Out
+    Expected: ExpectedOut
+DescriptorSets:
+  - Resources:    
+    - Name: Out
+      Kind: RWStructuredBuffer
+      DirectXBinding:
+        Register: 1
+        Space: 0
+      VulkanBinding:
+        Binding: 1
+...
+#--- end
+
+# REQUIRES: WaveSize_128
+
+# RUN: split-file %s %t
+# RUN: %dxc_target -T cs_6_6 -Fo %t.o %t/source.hlsl
+# RUN: %offloader %t/pipeline.yaml %t.o

test/WaveOps/WaveActiveBallot.Wave32.test

@@ -0,0 +1,60 @@
+#--- source.hlsl
+RWStructuredBuffer<uint4> Out : register(u1);
+
+[WaveSize(32)]
+[numthreads(32, 1, 1)]
+void main(uint3 threadID : SV_DispatchThreadID) {
+  // We expect the first resulting uint's bitmask to be 
+  // 0xBFFFFFFF
+  Out[0] = WaveActiveBallot(threadID.x != 30);
+
+  // We expect the first resulting uint's bitmask to be 
+  // 0x0000FFFF  
+  Out[1] = WaveActiveBallot(threadID.x < 16);
+
+  // We expect the resulting uint4 to be 4 0x00000000's
+  Out[2] = WaveActiveBallot(false);
+
+  // We expect the first resulting uint's bitmask to be 
+  // 0xAAAAAAAA
+  Out[3] = WaveActiveBallot(threadID.x % 2 == 1);
+}
+
+//--- pipeline.yaml
+
+---
+Shaders:
+  - Stage: Compute
+    Entry: main
+    DispatchSize: [1, 1, 1]
+Buffers:  
+  - Name: Out
+    Format: UInt32
+    Stride: 4
+    FillSize: 64
+  - Name: ExpectedOut
+    Format: UInt32
+    Stride: 4
+    Data: [3221225471, 0, 0, 0, 65535, 0, 0, 0, 0, 0, 0, 0, 2863311530, 0, 0, 0]
+Results:
+  - Result: Test
+    Rule: BufferExact
+    Actual: Out
+    Expected: ExpectedOut
+DescriptorSets:
+  - Resources:    
+    - Name: Out
+      Kind: RWStructuredBuffer
+      DirectXBinding:
+        Register: 1
+        Space: 0
+      VulkanBinding:
+        Binding: 1
+...
+#--- end
+
+# REQUIRES: WaveSize_32
+
+# RUN: split-file %s %t
+# RUN: %dxc_target -T cs_6_6 -Fo %t.o %t/source.hlsl
+# RUN: %offloader %t/pipeline.yaml %t.o

test/WaveOps/WaveActiveBallot.Wave64.test

@@ -0,0 +1,60 @@
+#--- source.hlsl
+RWStructuredBuffer<uint4> Out : register(u1);
+
+[WaveSize(64)]
+[numthreads(64, 1, 1)]
+void main(uint3 threadID : SV_DispatchThreadID) {
+  // We expect the first and second resulting uint's bitmask to be 
+  // 0xBFFFFFFF
+  Out[0] = WaveActiveBallot(threadID.x != 30 && threadID.x != 62);
+
+  // We expect the first resulting uint's bitmask to be 
+  // 0x0000FFFF, and the second to be 0xFFFF0000
+  Out[1] = WaveActiveBallot(threadID.x < 16 || threadID.x > 47 );
+
+  // We expect the resulting uint4 to be 4 0x00000000's
+  Out[2] = WaveActiveBallot(false);
+
+  // We expect the first and second resulting uint's bitmask to be 
+  // 0xAAAAAAAA
+  Out[3] = WaveActiveBallot(threadID.x % 2 == 1);
+}
+
+//--- pipeline.yaml
+
+---
+Shaders:
+  - Stage: Compute
+    Entry: main
+    DispatchSize: [1, 1, 1]
+Buffers:  
+  - Name: Out
+    Format: UInt32
+    Stride: 4
+    FillSize: 64
+  - Name: ExpectedOut
+    Format: UInt32
+    Stride: 4
+    Data: [3221225471, 3221225471, 0, 0, 65535, 4294901760, 0, 0, 0, 0, 0, 0, 2863311530, 2863311530, 0, 0]
+Results:
+  - Result: Test
+    Rule: BufferExact
+    Actual: Out
+    Expected: ExpectedOut
+DescriptorSets:
+  - Resources:    
+    - Name: Out
+      Kind: RWStructuredBuffer
+      DirectXBinding:
+        Register: 1
+        Space: 0
+      VulkanBinding:
+        Binding: 1
+...
+#--- end
+
+# REQUIRES: WaveSize_64
+
+# RUN: split-file %s %t
+# RUN: %dxc_target -T cs_6_6 -Fo %t.o %t/source.hlsl
+# RUN: %offloader %t/pipeline.yaml %t.o