summaryrefslogtreecommitdiffstats
path: root/tests
diff options
context:
space:
mode:
Diffstat (limited to 'tests')
-rw-r--r--tests/hlsl-intrinsic/wave-prefix-count-bits.slang17
-rw-r--r--tests/hlsl-intrinsic/wave-prefix-count-bits.slang.expected.txt8
2 files changed, 25 insertions, 0 deletions
diff --git a/tests/hlsl-intrinsic/wave-prefix-count-bits.slang b/tests/hlsl-intrinsic/wave-prefix-count-bits.slang
new file mode 100644
index 000000000..3aea7926d
--- /dev/null
+++ b/tests/hlsl-intrinsic/wave-prefix-count-bits.slang
@@ -0,0 +1,17 @@
+//TEST_CATEGORY(wave, compute)
+//DISABLE_TEST:COMPARE_COMPUTE_EX:-cpu -compute
+//DISABLE_TEST:COMPARE_COMPUTE_EX:-slang -compute
+//TEST:COMPARE_COMPUTE_EX:-slang -compute -dx12 -use-dxil -profile cs_6_0
+//TEST(vulkan):COMPARE_COMPUTE_EX:-vk -compute
+//TEST:COMPARE_COMPUTE_EX:-cuda -compute -render-features cuda_sm_7_0
+
+//TEST_INPUT:ubuffer(data=[0 0 0 0 0 0 0 0], stride=4):out,name outputBuffer
+RWStructuredBuffer<int> outputBuffer;
+
+[numthreads(8, 1, 1)]
+void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID)
+{
+ int idx = int(dispatchThreadID.x);
+
+ outputBuffer[idx] = WavePrefixCountBits(idx & 5);
+} \ No newline at end of file
diff --git a/tests/hlsl-intrinsic/wave-prefix-count-bits.slang.expected.txt b/tests/hlsl-intrinsic/wave-prefix-count-bits.slang.expected.txt
new file mode 100644
index 000000000..989297a9e
--- /dev/null
+++ b/tests/hlsl-intrinsic/wave-prefix-count-bits.slang.expected.txt
@@ -0,0 +1,8 @@
+0
+0
+1
+1
+2
+3
+4
+5