2 files changed, 10 insertions, 1 deletions
diff --git a/source/blender/draw/engines/compositor/compositor_shader.cc b/source/blender/draw/engines/compositor/compositor_shader.cc
index 3db6bdee6e8..d9e8e4b5f79 100644
--- a/source/blender/draw/engines/compositor/compositor_shader.cc
+++ b/source/blender/draw/engines/compositor/compositor_shader.cc
@@ -77,6 +77,14 @@ char *ShaderModule::pass_shader_code_frag_get(const GPUCodegenOutput *codegen,
   return DRW_shader_library_create_shader_string(shader_lib_, output.c_str());
 }
 
+char *ShaderModule::pass_shader_code_defs_get()
+{
+  std::string defines = "";
+  defines += "#define COMPOSITOR_SHADER\n";
+
+  return BLI_strdup(defines.c_str());
+}
+
 /* WATCH: This can be called from another thread! Needs to not touch the shader module in any
  * thread unsafe manner. */
 GPUShaderSource ShaderModule::pass_shader_code_generate(const GPUCodegenOutput *codegen,
@@ -86,7 +94,7 @@ GPUShaderSource ShaderModule::pass_shader_code_generate(const GPUCodegenOutput *
   source.vertex = pass_shader_code_vert_get(codegen, gpumat);
   source.fragment = pass_shader_code_frag_get(codegen, gpumat);
   source.geometry = nullptr;
-  source.defines = nullptr;
+  source.defines = pass_shader_code_defs_get();
   return source;
 }
 
diff --git a/source/blender/draw/engines/compositor/compositor_shader.hh b/source/blender/draw/engines/compositor/compositor_shader.hh
index ae7cd70eda9..4ab92534c41 100644
--- a/source/blender/draw/engines/compositor/compositor_shader.hh
+++ b/source/blender/draw/engines/compositor/compositor_shader.hh
@@ -40,6 +40,7 @@ class ShaderModule {
  private:
   char *pass_shader_code_vert_get(const GPUCodegenOutput *codegen, GPUMaterial *gpumat);
   char *pass_shader_code_frag_get(const GPUCodegenOutput *codegen, GPUMaterial *gpumat);
+  char *pass_shader_code_defs_get();
 };
 
 }  // namespace blender::compositor