diff options
| author | jsmall-nvidia <jsmall@nvidia.com> | 2020-03-10 12:31:25 -0400 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2020-03-10 12:31:25 -0400 |
| commit | a10d9cd8767e88a064719d71cc97144ba8b112d1 (patch) | |
| tree | c54745fb698c8cacfeb1c4440261eb899338f20e /tests | |
| parent | 721d2e8a2d457081cd3d9b081979d436b7002c2c (diff) | |
WIP Prefix Sum for CUDA (#1268)
* Fix some typos.
* Add wave-prefix-sum.slang test
* First pass at implementing prefixSum.
* Small improvments to prefixSum CUDA.
* Small improvement to prefix sum.
* Enable prefix sum in stdlib.
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/hlsl-intrinsic/wave-prefix-sum.slang | 16 | ||||
| -rw-r--r-- | tests/hlsl-intrinsic/wave-prefix-sum.slang.expected.txt | 8 |
2 files changed, 24 insertions, 0 deletions
diff --git a/tests/hlsl-intrinsic/wave-prefix-sum.slang b/tests/hlsl-intrinsic/wave-prefix-sum.slang new file mode 100644 index 000000000..f8d9bb560 --- /dev/null +++ b/tests/hlsl-intrinsic/wave-prefix-sum.slang @@ -0,0 +1,16 @@ +//DISABLE_TEST(compute):COMPARE_COMPUTE_EX:-cpu -compute +//DISABLE_TEST(compute):COMPARE_COMPUTE_EX:-slang -compute +//TEST(compute):COMPARE_COMPUTE_EX:-slang -compute -dx12 -use-dxil -profile cs_6_0 +//TEST(compute, vulkan):COMPARE_COMPUTE_EX:-vk -compute +//TEST(compute):COMPARE_COMPUTE_EX:-cuda -compute + +//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); + int val = WavePrefixSum(1 << idx); + outputBuffer[idx] = val; +}
\ No newline at end of file diff --git a/tests/hlsl-intrinsic/wave-prefix-sum.slang.expected.txt b/tests/hlsl-intrinsic/wave-prefix-sum.slang.expected.txt new file mode 100644 index 000000000..6ec6deeea --- /dev/null +++ b/tests/hlsl-intrinsic/wave-prefix-sum.slang.expected.txt @@ -0,0 +1,8 @@ +0 +1 +3 +7 +F +1F +3F +7F |