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

This commit is contained in:
BeastLe9enD 2022-04-24 20:01:33 +02:00 committed by Eduard-Mihai Burtescu
parent 685c79a972
commit 4b55f91de3

View File

@ -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)]'",
),
));
}