Support compute shaders and compute operations

[libs/gl.git] / source / core / program.cpp

diff --git a/source/core/program.cpp b/source/core/program.cpp
index 8ac13bbd0ce21454cb5af1b269dacc0fc6770205..91f215651b57df64b834315c13aa89b86d57d546 100644 (file)
--- a/source/core/program.cpp
+++ b/source/core/program.cpp
@@ -56,6 +56,10 @@ void Program::add_stages(const Module &mod, const map<string, int> &spec_values)
                collect_uniforms(spirv_mod);
                collect_attributes(spirv_mod);
                collect_builtins(spirv_mod);
+
+               for(const SpirVModule::EntryPoint &e: spirv_mod.get_entry_points())
+                       if(e.stage==SpirVModule::COMPUTE)
+                               reflect_data.compute_wg_size = e.compute_local_size;
        }
 
        finalize_uniforms();