spirv/tests: Parametrize stage in get_nir() helper

Reviewed-by: Jesse Natalie <jenatali@microsoft.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/21512>
This commit is contained in:
Caio Oliveira
2023-02-23 19:40:47 -08:00
committed by Marge Bot
parent 131f328a18
commit 4e5b520286

View File

@@ -41,7 +41,7 @@ protected:
glsl_type_singleton_decref();
}
void get_nir(size_t num_words, const uint32_t *words)
void get_nir(size_t num_words, const uint32_t *words, gl_shader_stage stage = MESA_SHADER_COMPUTE)
{
spirv_to_nir_options spirv_options;
memset(&spirv_options, 0, sizeof(spirv_options));
@@ -60,7 +60,7 @@ protected:
nir_options.use_scoped_barrier = true;
shader = spirv_to_nir(words, num_words, NULL, 0,
MESA_SHADER_COMPUTE, "main", &spirv_options, &nir_options);
stage, "main", &spirv_options, &nir_options);
}
nir_intrinsic_instr *find_intrinsic(nir_intrinsic_op op, unsigned index=0)