summaryrefslogtreecommitdiff
path: root/tests/metal
diff options
context:
space:
mode:
Diffstat (limited to 'tests/metal')
-rw-r--r--tests/metal/thread_position_in_threadgroup.slang27
1 files changed, 27 insertions, 0 deletions
diff --git a/tests/metal/thread_position_in_threadgroup.slang b/tests/metal/thread_position_in_threadgroup.slang
new file mode 100644
index 000000000..a20fe2ec7
--- /dev/null
+++ b/tests/metal/thread_position_in_threadgroup.slang
@@ -0,0 +1,27 @@
+//TEST:SIMPLE(filecheck=CHECK): -entry computeMain -stage compute -target metal -D GROUPID
+//TEST:SIMPLE(filecheck=CHECK): -entry computeMain -stage compute -target metal
+
+//CHECK: computeMain
+
+// ensure we compute the SV_GroupIndex from SV_GroupThreadID and `numthreads`
+// CHECK: thread_position_in_threadgroup
+// CHECK-DAG: *{{.*}}2
+
+RWBuffer<uint> dst;
+
+void indirection(uint groupIndex)
+{
+ dst[groupIndex] = groupIndex;
+}
+
+#define THREAD_COUNT 2
+[numthreads(THREAD_COUNT, 1, 1)]
+#ifdef GROUPID
+void computeMain(uint GI : SV_GroupIndex, uint GTID : SV_GroupThreadID)
+#else
+void computeMain(uint GI : SV_GroupIndex)
+#endif
+{
+ dst[GI + THREAD_COUNT] = GI;
+ indirection(GI);
+} \ No newline at end of file