summaryrefslogtreecommitdiff
path: root/tests/cuda/cuda-layout.slang
diff options
context:
space:
mode:
authorjsmall-nvidia <jsmall@nvidia.com>2021-06-10 14:57:09 -0400
committerGitHub <noreply@github.com>2021-06-10 11:57:09 -0700
commit37e8917d10626b519470f2e34625f0efe741352f (patch)
tree4e8e51bd63ebc03fcdf0c893b675906cf507d73c /tests/cuda/cuda-layout.slang
parent0d9bd79e8fd4d57e1a723ca6b6a45efec2b42872 (diff)
CUDA layout corner cases/testing (#1881)
* #include an absolute path didn't work - because paths were taken to always be relative. * Add support for sizeOf/alignOf/offsetOf to stdlib. Add $G intrinsic expansion that works of the generic parameters not the param type * Test cuda layout. * Fix CUDA layout issues. Fix reflection to handle other built in types. Fix __offsetOf * Tests of reflection and layout as reported directly from CUDA. * Comment about use of aligned size as size. * Fix warning from VS. * Check alignment is pow2. * Small improvements to alignment calcs. * Tab to spaces. * Fix alignment pointer sizes on 32 bit OS for CUDA. * Fix CUDA reflection on 32 bit.
Diffstat (limited to 'tests/cuda/cuda-layout.slang')
-rw-r--r--tests/cuda/cuda-layout.slang24
1 files changed, 24 insertions, 0 deletions
diff --git a/tests/cuda/cuda-layout.slang b/tests/cuda/cuda-layout.slang
new file mode 100644
index 000000000..725bf798e
--- /dev/null
+++ b/tests/cuda/cuda-layout.slang
@@ -0,0 +1,24 @@
+//TEST(compute):COMPARE_COMPUTE_EX:-cuda -compute -output-using-type -shaderobj
+
+//TEST_INPUT:ubuffer(data=[0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0], stride=4):out,name=outputBuffer
+RWStructuredBuffer<int> outputBuffer : register(u0);
+
+#define WRITE_TYPE_ALIGN(base, type) \
+ outputBuffer[base * 4 + 0] = __alignOf<type>(); \
+ outputBuffer[base * 4 + 1] = __alignOf<vector<type, 2> >(); \
+ outputBuffer[base * 4 + 2] = __alignOf<vector<type, 3> >(); \
+ outputBuffer[base * 4 + 3] = __alignOf<vector<type, 4> >();
+
+
+[numthreads(1, 1, 1)]
+void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID)
+{
+ WRITE_TYPE_ALIGN(0, uint8_t)
+ WRITE_TYPE_ALIGN(1, uint16_t)
+ WRITE_TYPE_ALIGN(2, int)
+ WRITE_TYPE_ALIGN(3, int64_t)
+
+ WRITE_TYPE_ALIGN(4, half)
+ WRITE_TYPE_ALIGN(5, float)
+ WRITE_TYPE_ALIGN(6, double)
+}