Make verify_grad don't mix float32 and float64