(1) update auto-casting api, (2) simplifiy precision arguments and re…
I have verified that this fix gives the same results as the baseline code with both fp16 and bf16.

Signed-off-by: Sangkug Lym <slym@nvidia.com>