Support 'threads' with 'mesh_nv' and 'task_nv'

Author: BeastLe9enD

crates/rustc_codegen_spirv/src/symbols.rs

@@ -651,7 +651,7 @@ fn parse_entry_attrs(
                 .execution_modes
                 .push((origin_mode, ExecutionModeExtra::new([])));
         }
-        GLCompute => {
+        GLCompute | MeshNV | TaskNV => {
             if let Some(local_size) = local_size {
                 entry
                     .execution_modes
@@ -660,7 +660,7 @@ fn parse_entry_attrs(
                 return Err((
                     arg.span(),
                     String::from(
-                        "The `threads` argument must be specified when using `#[spirv(compute)]`",
+                        "The `threads` argument must be specified when using `#[spirv(compute)]`, '#[spirv(mesh_nv)]' or '#[spirv(task_nv)]'",
                     ),
                 ));
             }