fix updates' float32 bug

use T.cast to manully set updates' type