Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Remove instanceNormalization op from DEFAULT_OP_BLOCK_LIST (#202)
This PR is to remove InstanceNormalization op from DEFAULT_OP_BLOCK_LIST. The background is: - 3 months ago, I found InstanceNormalization was unable to work on fp16. It would generate below error: ``` File "/usr/local/lib/python3.6/dist-packages/onnxruntime/capi/onnxruntime_inference_collection.py", line 124, in run return self._sess.run(output_names, input_feed, run_options) onnxruntime.capi.onnxruntime_pybind11_state.Fail: [ONNXRuntimeError] : 1 : FAIL : Non-zero status code returned while running InstanceNormalization node. Name:'InstanceNormalization_31' Status Message: CUDNN error executing cudnnBatchNormalizationForwardTraining( CudnnHandle(), CUDNN_BATCHNORM_SPATIAL, &one, &zero, data_desc, x_data, data_desc, y_data, stats_desc, unused_scale.get(), unused_bias.get(), 1.0f, mean.get(), variance.get(), CUDNN_BN_MIN_EPSILON, nullptr, nullptr) ``` And, @TomWildenhain-Microsoft helped create this PR to block InstanceNormalization to convert fp16: #197 - Meanwhile, I also asked ORT team's help to investigate and fix InstanceNormalization op issue on fp16. The fix was done recently (Nov. 29, 2021) by @hariharans29 in this PR: microsoft/onnxruntime#9879 So, InstanceNormalization is no needed to be blocked any more. And this PR is to remove InstanceNormalization op from DEFAULT_OP_BLOCK_LIST.
- Loading branch information