Skip to content

Commit c601948

Browse files
committed
Build issue
1 parent b9fe441 commit c601948

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

orttraining/orttraining/core/optimizer/qdq_fusion.cc

+8-2
Original file line numberDiff line numberDiff line change
@@ -29,8 +29,14 @@ int ReplaceOrCreateZeroPointInitializer(Graph& graph, Node& quantize_node) {
2929
zero_point_type = zero_point_tensor_int.data_type();
3030
zero_point_tensor_float.set_name(graph.GenerateNodeArgName(zero_point_tensor_int.name()));
3131
zero_point_tensor_float.set_data_type(ONNX_NAMESPACE::TensorProto_DataType_FLOAT);
32-
for (const auto val : zero_point_tensor_int.DataAsSpan<int32_t>()) {
33-
zero_point_tensor_float.add_float_data(static_cast<float>(val));
32+
if (zero_point_type == ONNX_NAMESPACE::TensorProto_DataType_UINT8) {
33+
for (const auto val : zero_point_tensor_int.DataAsSpan<uint8_t>()) {
34+
zero_point_tensor_float.add_float_data(static_cast<float>(val));
35+
}
36+
} else {
37+
for (const auto val : zero_point_tensor_int.DataAsSpan<int8_t>()) {
38+
zero_point_tensor_float.add_float_data(static_cast<float>(val));
39+
}
3440
}
3541
for (const auto dim : zero_point_tensor_int.dims()) {
3642
zero_point_tensor_float.add_dims(dim);

0 commit comments

Comments
 (0)