fix one bug in cpu version