diff --git a/crates/rustc_codegen_spirv/src/symbols.rs b/crates/rustc_codegen_spirv/src/symbols.rs index 16df6c713c..35e5b06fa1 100644 --- a/crates/rustc_codegen_spirv/src/symbols.rs +++ b/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)]'", ), )); }