X-Git-Url: http://git.tdb.fi/?p=libs%2Fgl.git;a=blobdiff_plain;f=source%2Fcore%2Fprogram.cpp;h=91f215651b57df64b834315c13aa89b86d57d546;hp=8ac13bbd0ce21454cb5af1b269dacc0fc6770205;hb=cebf1330ef6773b7b4496dc279ec02a7ca4351bb;hpb=2a70fecfbbe8708be2bdaa75d222dd5a889a8ed3

diff --git a/source/core/program.cpp b/source/core/program.cpp
index 8ac13bbd..91f21565 100644
--- 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();