1 files changed, 8 insertions, 1 deletions
diff --git a/src/shader_recompiler/backend/spirv/emit_spirv_context_get_set.cpp b/src/shader_recompiler/backend/spirv/emit_spirv_context_get_set.cpp
index feca5105f..64f828107 100644
--- a/src/shader_recompiler/backend/spirv/emit_spirv_context_get_set.cpp
+++ b/src/shader_recompiler/backend/spirv/emit_spirv_context_get_set.cpp
@@ -549,8 +549,15 @@ Id EmitInvocationInfo(EmitContext& ctx) {
     case Stage::TessellationEval:
         return ctx.OpShiftLeftLogical(ctx.U32[1], ctx.OpLoad(ctx.U32[1], ctx.patch_vertices_in),
                                       ctx.Const(16u));
+    case Stage::Fragment:
+        // Return sample mask in upper 16 bits
+        return ctx.OpShiftLeftLogical(ctx.U32[1], ctx.OpLoad(ctx.U32[1], ctx.sample_mask),
+                                      ctx.Const(16u));
+    case Stage::Compute:
+        // For compute shaders, return standard format since we can't access workgroup size directly
+        return ctx.Const(0x00ff0000u);
     default:
-        LOG_WARNING(Shader, "(STUBBED) called");
+        // For other stages, return the standard invocation info format
         return ctx.Const(0x00ff0000u);
     }
 }