2 files changed, 7 insertions, 6 deletions
diff --git a/tests/ir/loop.slang b/tests/ir/loop.slang
index d637e4536..acdba0b20 100644
--- a/tests/ir/loop.slang
+++ b/tests/ir/loop.slang
@@ -1,4 +1,4 @@
-//TEST:SIMPLE:-target slang-ir-assembly -profile cs_4_0 -entry main
+//TEST:SIMPLE:-use-ir -dump-ir -skip-codegen -target hlsl -profile cs_5_0 -entry main
 
 #define GROUP_THREAD_COUNT 64
 
@@ -9,8 +9,8 @@ groupshared float4 s[GROUP_THREAD_COUNT];
 
 [numthreads(GROUP_THREAD_COUNT, 1, 1)]
 void main(
-    uint dispatchThreadID   : SV_DispatchThreadIndex,
-    uint groupThreadID      : SV_GroupThreadIndex,
+    uint dispatchThreadID   : SV_DispatchThreadID,
+    uint groupThreadID      : SV_GroupThreadID,
     uint groupID            : SV_GroupIndex )
 {
     // the actual algorithm being done here is bogus
diff --git a/tests/ir/loop.slang.expected b/tests/ir/loop.slang.expected
index e34bb68cc..a9122c094 100644
--- a/tests/ir/loop.slang.expected
+++ b/tests/ir/loop.slang.expected
@@ -1,9 +1,7 @@
 result code = 0
 standard error = {
-}
-standard output = {
 
-ir_global_var %1	: Ptr<vector<float,4>[64]>;
+ir_global_var %1	: Ptr<@ThreadGroup vector<float,4>[64]>;
 
 ir_global_var %2	: Ptr<StructuredBuffer<vector<float,4>>>;
 
@@ -78,4 +76,7 @@ block %19:
 	bufferStore(%43, %44, %46)
 	return_void()
 }
+
+}
+standard output = {
 }