define WGPU

bevyengine · Nov 12, 2020 · eb7bf0b · eb7bf0b
1 parent 434a3e6
commit eb7bf0b
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/crates/bevy_wgpu/src/renderer/wgpu_render_resource_context.rs b/crates/bevy_wgpu/src/renderer/wgpu_render_resource_context.rs
@@ -570,9 +570,15 @@ impl RenderResourceContext for WgpuRenderResourceContext {
     }
 
     fn get_specialized_shader(&self, shader: &Shader, macros: Option<&[String]>) -> Shader {
+        let macros: Vec<String> = macros
+            .unwrap_or(&[])
+            .iter()
+            .chain((&["WGPU".to_string()]).iter())
+            .cloned()
+            .collect();
         let spirv_data = match shader.source {
             ShaderSource::Spirv(ref bytes) => bytes.clone(),
-            ShaderSource::Glsl(ref source) => glsl_to_spirv(&source, shader.stage, macros),
+            ShaderSource::Glsl(ref source) => glsl_to_spirv(&source, shader.stage, Some(&macros)),
         };
         Shader {
             source: ShaderSource::Spirv(spirv_data),