fix: fix export test value bug.

espressif · Dec 23, 2024 · 10d6d37 · 10d6d37
1 parent 5428134
commit 10d6d37
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/ppq/parser/espdl/helper.py b/ppq/parser/espdl/helper.py
@@ -170,7 +170,7 @@ def make_graph_test_value(
                 input_value = input_value.flatten()
             if len(input_shape) == 0:
                 input_shape = [1]
-            input_value_fbs = make_tensor(input_name, tensor_type, input_shape, input_value.tobytes(), raw=True,
+            input_value_fbs = make_tensor(input_name, tensor_type, input_shape, input_value, raw=True,
                                           exponents=var_exponents)
             input_test_values.append(input_value_fbs)
 
@@ -183,7 +183,7 @@ def make_graph_test_value(
                 output_value = output_value.flatten()
             if len(output_shape) == 0:
                 output_shape = [1]
-            output_value_fbs = make_tensor(output_name, tensor_type, output_shape, output_value.tobytes(), raw=True,
+            output_value_fbs = make_tensor(output_name, tensor_type, output_shape, output_value, raw=True,
                                            exponents=var_exponents)
             output_test_values.append(output_value_fbs)
 

diff --git a/ppq/parser/espdl_exporter.py b/ppq/parser/espdl_exporter.py
@@ -441,6 +441,7 @@ def build_variable_proto(
                         value.item(),
                     ]  # it is fine for onnx, shape for this value will be []
                     var_shape = torch.tensor(value).shape
+                    is_raw_format = True
             else:
                 value = value  # value is python primary type.
             tensor_proto = helper.make_tensor(