fix the mismatch running on cpu devices