fix difference of colormap behavior across different GPUs