使用ModernGL初始化Compute Shader时半数Uniform丢失的问题

我使用Python的ModernGL库开发辐射传输相关的Compute Shader，创建Shader对象后发现，Shader代码中声明的Uniform并未全部出现在对象中，近半数Uniform丢失，且无论调整声明顺序还是显式指定值类型，丢失的始终是同一批。

相关代码片段 #

Python代码 #

import moderngl as mgl

# 创建上下文
context = mgl.create_standalone_context(require=440)
# 使用外部文件中的源码创建Compute Shader
compute_shader = context.compute_shader(src_code)

# 为预定义字典中的Uniform赋值，字典键与Shader中Uniform名称匹配
for key, value in uniforms.items():
    compute_shader.get(key, default=None).write(value)

Shader源码（头部） #

#version 440

uniform float instrument_azimut;
uniform float instrument_elevation;
uniform float instrument_altitude;
uniform float instrument_lon;
uniform float instrument_lat;
uniform float instrument_area;
uniform float instrument_fov;
uniform float wavelength;
uniform int   los_nb_points;
uniform float map_delta_az;
uniform bool  is_point_source;
uniform int   atm_nb_altitudes;
uniform int   atm_nb_angles;
uniform bool  use_aerosol;
// 可迭代Uniform
uniform float instrument_los[3];
uniform float los_altitudes[100];
uniform float atm_altitudes[121];
uniform float map_azimuts[10];
uniform float map_distances[10];
uniform float los_volumes[100];
uniform float atm_total_abs[121];
uniform float atm_ray_beta[121];
uniform float atm_aer_beta[121];
uniform float atm_angles[100];
uniform float atm_aer_Pfct[121];
uniform float atm_aer_Pfct_DoLP[121];
uniform float los_transmittance[100];


#define X 1
#define Y 1
#define Z 100


// 设置计算工作组
layout(local_size_x = X, local_size_y = Y, local_size_z = Z) in;

// 定义常量值
const float EARTH_RADIUS = 6371.; //km
const float PI = 3.14159;
const float DtoR = PI / 180;
const float RtoD = 1. / DtoR;


// Stokes参数缓冲区
layout(std430, binding=0) buffer in_0{
    float in_V[];};
layout(std430, binding=1) buffer in_1{
    float in_Vcos[];};
layout(std430, binding=2) buffer in_2{
    float in_Vsin[];};

layout(std430, binding=3) buffer out_3{
    float out_V[];};
layout(std430, binding=4) buffer out_4{
    float out_Vcos[];};
layout(std430, binding=5) buffer out_5{
    float out_Vsin[];};

// ...所有实用函数...

丢失的Uniform情况 #

遍历Compute Shader对象输出的Uniform列表如下，缺失了instrument_azimut、instrument_lon、instrument_lat、instrument_fov、wavelength、los_nb_points、instrument_los等：

instrument_elevation
instrument_altitude
instrument_area
map_delta_az
is_point_source
atm_nb_altitudes
atm_nb_angles
use_aerosol
los_altitudes
atm_altitudes
map_azimuts
map_distances
los_volumes
atm_total_abs
atm_ray_beta
atm_aer_beta
atm_angles
atm_aer_Pfct
atm_aer_Pfct_DoLP
los_transmittance

问题原因与解决方案 #

核心原因 #

GLSL编译器会自动优化移除未被实际使用的Uniform。如果某个Uniform仅在Shader中声明，但未在main函数或被main调用的任何函数中读取或使用，编译器会将其从最终的Shader程序中剔除，导致ModernGL无法获取到这些Uniform。

解决步骤 #

1. 检查Uniform的实际使用情况
   逐一确认缺失的Uniform是否在Shader的执行逻辑中被引用。比如instrument_lon是否在实用函数或main函数中有读取操作，instrument_los数组是否有被访问的代码。

2. 强制保留未使用的Uniform
   如果某些Uniform需要保留（比如后续会扩展逻辑用到，或需要占位），可以在main函数中添加“伪使用”代码，让编译器认为变量被使用：

   void main() {
       // 伪使用未被引用的Uniform，避免被编译器优化移除
       (void)instrument_azimut;
       (void)instrument_lon;
       (void)instrument_lat;
       (void)instrument_fov;
       (void)wavelength;
       (void)los_nb_points;
       (void)instrument_los;
       
       // ... 原有的main函数逻辑
   }
   

3. 查看Shader编译日志
   通过ModernGL获取Shader的编译日志，确认哪些Uniform被编译器优化移除：

   print(compute_shader.info)
   

   日志中通常会包含关于未使用变量被剔除的提示，帮助你定位问题。

内容的提问来源于stack exchange，提问作者Azireo