2 files changed, 6 insertions, 5 deletions
diff --git a/tests/language-feature/descriptor-handle/desc-handle-default.slang b/tests/language-feature/descriptor-handle/desc-handle-default.slang
index 016b04bfb..9a4d444d7 100644
--- a/tests/language-feature/descriptor-handle/desc-handle-default.slang
+++ b/tests/language-feature/descriptor-handle/desc-handle-default.slang
@@ -109,12 +109,13 @@ uniform RWStructuredBuffer<float>.Handle rwBuffer;
 uniform RaytracingAccelerationStructure.Handle rayAcceleration;
 #endif
 
+[numthreads(2,2,1)]
 [shader("compute")]
 void computeMain()
 {
 
     t1[0] = t2[0] + t2[0] + t4[0];
-    
+
 #ifdef SAMPLER
     t1[2] = t4.Sample(sampler, 0);
 #endif
@@ -148,7 +149,7 @@ void computeMain()
 #ifdef STORAGE_BUFFER
     t1[0] += rwBuffer[0];
 #endif
-  
+
 #ifdef ACCELERATION_STRUCTURE
     RayDesc ray;
     ray.Origin = float3(0.1f, 0.1f, 0.0f);
@@ -162,4 +163,4 @@ void computeMain()
     rq.Abort();
     t1[13] = (float)rq.RayFlags();
 #endif
-}
-\ No newline at end of file
+}
diff --git a/tests/language-feature/descriptor-handle/desc-handle-vk-mutable-descriptor.slang b/tests/language-feature/descriptor-handle/desc-handle-vk-mutable-descriptor.slang
index 2bf57d0ef..93981cd27 100644
--- a/tests/language-feature/descriptor-handle/desc-handle-vk-mutable-descriptor.slang
+++ b/tests/language-feature/descriptor-handle/desc-handle-vk-mutable-descriptor.slang
@@ -51,7 +51,7 @@ uniform StructuredBuffer<float>.Handle buffer2;
 uniform RWStructuredBuffer<float>.Handle rwBuffer;
 uniform RaytracingAccelerationStructure.Handle rayAcceleration;
 
-
+[numthreads(2,2,1)]
 [shader("compute")]
 void computeMain()
 {
@@ -80,4 +80,4 @@ void computeMain()
     rq.Abort();
     t1[13] = (float)rq.RayFlags();
 
-}
-\ No newline at end of file
+}