PyOpenGL Compute Shader无法写入纹理的排查与验证求助

我正在基于Python实现光线追踪器的Compute Shader版本，参考learnopengl的Compute Shader教程。目前无法让Compute Shader写入纹理，甚至无法确认Shader是否正常运行，请问可通过哪些测试验证其编译与运行状态？

纹理及显示相关代码功能正常，直接上传图像到纹理可正常显示。以下是相关代码：

SETUP 代码 #

self.compute_shader = glCreateShader(GL_COMPUTE_SHADER)
text = open("../shaders/compute.glsl", 'r')
compute_shader_source = text.read()

glShaderSource(self.compute_shader, compute_shader_source)
glCompileShader(self.compute_shader)

self.program_id = glCreateProgram()
glAttachShader(self.program_id, self.compute_shader)
glLinkProgram(self.program_id)

self.texture = 0
glGenTextures(1, self.texture)
glActiveTexture(GL_TEXTURE0)
glBindTexture(GL_TEXTURE_2D, self.texture)

glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_EDGE)
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_CLAMP_TO_EDGE)
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR)
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR)

glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA32F, 200, 160, 0, GL_RGBA, GL_UNSIGNED_BYTE, None)
glBindImageTexture(0, self.texture, 0, GL_FALSE, 0, GL_READ_WRITE, GL_RGBA32F)

RENDER CALL 代码 #

glActiveTexture(GL_TEXTURE0)
glBindTexture(GL_TEXTURE_2D, self.texture)

glUseProgram(self.program_id)
glDispatchCompute(200, 160, 1)
glMemoryBarrier(GL_ALL_BARRIER_BITS)

COMPUTE SHADER 代码 #

#version 430 core

layout (local_size_x = 1, local_size_y = 1, local_size_z = 1) in;

layout (rgba32f, binding = 0) uniform image2D imgOutput;

void main() {
    vec4 value = vec4(1.0, 0.0, 0.0, 1.0);
    ivec2 texelCoord = ivec2(gl_GlobalInvocationID.xy);
    
    value.x = float(texelCoord.x)/(gl_NumWorkGroups.x);
    value.y = float(texelCoord.y)/(gl_NumWorkGroups.y);
    
    imageStore(imgOutput, texelCoord, value);
}

运行代码后仅显示黑屏，加载图像到纹理后，Compute Shader也未覆盖原有图像，推测Shader未运行或未写入纹理。（屏幕与纹理尺寸均为200×160）

1. 检查Shader编译错误 #

在glCompileShader调用后立即查询编译状态和错误日志，这是排查Shader问题的第一步：

compiled = glGetShaderiv(self.compute_shader, GL_COMPILE_STATUS)
if not compiled:
    error_log = glGetShaderInfoLog(self.compute_shader)
    print("Compute Shader编译错误:\n", error_log.decode('utf-8'))

2. 检查程序链接错误 #

Shader编译通过不代表程序链接正常，在glLinkProgram后检查链接状态：

linked = glGetProgramiv(self.program_id, GL_LINK_STATUS)
if not linked:
    error_log = glGetProgramInfoLog(self.program_id)
    print("程序链接错误:\n", error_log.decode('utf-8'))

3. 简化Shader逻辑，输出固定颜色 #

去掉动态计算逻辑，让Shader直接写入纯红色，排除计算逻辑导致的问题：

#version 430 core

layout (local_size_x = 1, local_size_y = 1, local_size_z = 1) in;

layout (rgba32f, binding = 0) uniform image2D imgOutput;

void main() {
    ivec2 texelCoord = ivec2(gl_GlobalInvocationID.xy);
    imageStore(imgOutput, texelCoord, vec4(1.0, 0.0, 0.0, 1.0));
}

如果仍显示黑屏，说明Shader未执行或纹理绑定异常；如果显示红色，说明原Shader的动态计算逻辑存在问题（比如gl_NumWorkGroups的使用）。

4. 验证工作组与纹理坐标匹配 #

当前glDispatchCompute(200, 160, 1)对应200×160个工作组，每个工作组1个线程，gl_GlobalInvocationID.xy范围(0-199, 0-159)，和纹理尺寸匹配。可尝试调整工作组大小（比如local_size_x=16, local_size_y=16），然后修改调度参数为(ceil(200/16), ceil(160/16), 1)，测试是否能正常写入。

5. 检查纹理绑定与格式一致性 #

确认以下参数匹配：

• glBindImageTexture的绑定点0与Shader中binding=0一致
• 纹理内部格式GL_RGBA32F与Shader中rgba32f匹配
• 访问模式GL_READ_WRITE设置正确

另外，glTexImage2D最后参数为None会导致纹理初始数据未定义，可尝试初始化纯色纹理后再测试Compute Shader是否覆盖。

6. 使用GPU调试工具 #

用RenderDoc、NVIDIA Nsight等工具捕获帧，查看：

• Compute Shader是否被调度执行
• 纹理内容是否被修改
• Shader内变量值是否符合预期

7. 确认OpenGL上下文版本 #

Shader使用#version 430 core，需确保上下文支持OpenGL 4.3及以上版本：

print("OpenGL版本:", glGetString(GL_VERSION).decode('utf-8'))

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