From bfbf7e0f24c290582569c3338a62f39bef3ea77d Mon Sep 17 00:00:00 2001
From: IceSentry <c.giguere42@gmail.com>
Date: Sat, 24 Sep 2022 01:22:47 -0400
Subject: [PATCH] add support for .comp glsl shaders

---
 crates/bevy_render/src/render_resource/shader.rs | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/crates/bevy_render/src/render_resource/shader.rs b/crates/bevy_render/src/render_resource/shader.rs
index 5913d44dd3aab..beee76e6f25a5 100644
--- a/crates/bevy_render/src/render_resource/shader.rs
+++ b/crates/bevy_render/src/render_resource/shader.rs
@@ -254,6 +254,10 @@ impl AssetLoader for ShaderLoader {
                     String::from_utf8(Vec::from(bytes))?,
                     naga::ShaderStage::Fragment,
                 ),
+                "comp" => Shader::from_glsl(
+                    String::from_utf8(Vec::from(bytes))?,
+                    naga::ShaderStage::Compute,
+                ),
                 _ => panic!("unhandled extension: {}", ext),
             };
 
@@ -279,7 +283,7 @@ impl AssetLoader for ShaderLoader {
     }
 
     fn extensions(&self) -> &[&str] {
-        &["spv", "wgsl", "vert", "frag"]
+        &["spv", "wgsl", "vert", "frag", "comp"]
     }
 }