For these cases, you should use full fp32 precision by explicitly disabling the default fp16 mode.