glasm: Declare NV_shader_thread_group when needed

This commit is contained in:
ReinUsesLisp 2021-05-14 02:09:33 -03:00 committed by ameerj
parent 568d813eea
commit d4385c34e3
1 changed files with 4 additions and 3 deletions

View File

@ -218,15 +218,16 @@ void SetupOptions(std::string& header, Info info) {
if (info.uses_atomic_f16x2_add || info.uses_atomic_f16x2_min || info.uses_atomic_f16x2_max) { if (info.uses_atomic_f16x2_add || info.uses_atomic_f16x2_min || info.uses_atomic_f16x2_max) {
header += "OPTION NV_shader_atomic_fp16_vector;"; header += "OPTION NV_shader_atomic_fp16_vector;";
} }
if (info.uses_subgroup_invocation_id || info.uses_subgroup_mask) { if (info.uses_subgroup_invocation_id || info.uses_subgroup_mask || info.uses_subgroup_vote) {
header += "OPTION NV_shader_thread_group;"; header += "OPTION NV_shader_thread_group;";
} }
if (info.uses_subgroup_shuffles) { if (info.uses_subgroup_shuffles) {
header += "OPTION NV_shader_thread_shuffle;"; header += "OPTION NV_shader_thread_shuffle;";
} }
// TODO: Track the shared atomic ops // TODO: Track the shared atomic ops
header += header += "OPTION NV_shader_storage_buffer;"
"OPTION NV_shader_storage_buffer;OPTION NV_gpu_program_fp64;OPTION NV_bindless_texture;"; "OPTION NV_gpu_program_fp64;"
"OPTION NV_bindless_texture;";
} }
} // Anonymous namespace } // Anonymous namespace