The other issue may be related to using fp16.