From 5e6cfd73a07812431c8da910f37f213857102acd Mon Sep 17 00:00:00 2001
From: Jake Turner <jake@evansturner.co.uk>
Date: Sun, 6 Apr 2025 12:25:05 +0100
Subject: [PATCH] Add D3D12_Workgroup_Zoo unit tests for workgroup query
 functions

WaveGetLaneCount()
WaveGetLaneIndex()
WaveIsFirstLane()
---
 util/test/demos/d3d12/d3d12_subgroup_zoo.cpp | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/util/test/demos/d3d12/d3d12_subgroup_zoo.cpp b/util/test/demos/d3d12/d3d12_subgroup_zoo.cpp
index e7d5f2010..c020bca59 100644
--- a/util/test/demos/d3d12/d3d12_subgroup_zoo.cpp
+++ b/util/test/demos/d3d12/d3d12_subgroup_zoo.cpp
@@ -252,6 +252,13 @@ void main(uint3 inTid : SV_DispatchThreadID)
       data.x += WaveActiveSum(id);
     }
   }
+  else if(IsTest(9))
+  {
+    // Query function : unit tests
+    data.x = float(WaveGetLaneCount());
+    data.y = float(WaveGetLaneIndex());
+    data.z = float(WaveIsFirstLane());
+  }
   SetOuput(data);
 }