From 78133434b58337aef536aeb06b248deccf9f564a Mon Sep 17 00:00:00 2001 From: Ye Wang <52801275+wangyems@users.noreply.github.com> Date: Thu, 17 Mar 2022 22:38:43 -0700 Subject: [PATCH] Fix fp16 converter bugs[1/n] (#10882) handle sequence type --- onnxruntime/python/tools/transformers/float16.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/onnxruntime/python/tools/transformers/float16.py b/onnxruntime/python/tools/transformers/float16.py index 3823e001e0..b5b26c0b04 100644 --- a/onnxruntime/python/tools/transformers/float16.py +++ b/onnxruntime/python/tools/transformers/float16.py @@ -280,6 +280,12 @@ def convert_float_to_float16(model, if n.name not in graph_io_to_skip: n.type.tensor_type.elem_type = onnx_proto.TensorProto.FLOAT16 value_info_list.append(n) + if n.type.HasField('sequence_type'): + if n.type.sequence_type.elem_type.tensor_type.elem_type == onnx_proto.TensorProto.FLOAT: + if n.name not in graph_io_to_skip: + n.type.sequence_type.elem_type.tensor_type.elem_type = onnx_proto.TensorProto.FLOAT16 + value_info_list.append(n) + queue = next_level for key, value in fp32_initializers.items():