Fix the mismatch on NV GPUs