Fixed compute shader for NVIDIA.