fix load_and_run.py

Author: tangzhiyi11

dicp/dicp/vendor/AscendGraph/codegen/load_and_run.py

@@ -209,14 +209,18 @@ def __init__(self, graph_id, device_id):
         self.output_dtypes = []
         self.output_datasize = []
         for item in shapes:
+            if item == '':
+                self.output_shapes.append([])
+                continue
             elems = item.split(',')
             elems = [int(x) for x in elems]
             self.output_shapes.append(elems)
         for item in dtypes:
             elem = int(item)
             self.output_dtypes.append(elem)
         for i in range(len(shapes)):
-            elem_size = math.prod(self.output_shapes[i])
+            elem_size = math.prod(self.output_shapes[i]) if len(
+                self.output_shapes[i]) > 0 else 1
             self.output_datasize.append(
                 elem_size * acl.data_type_size(self.output_dtypes[i]))
         self.output_datasize_c = (
@@ -242,14 +246,18 @@ def __init__(self, graph_id, device_id):
         self.input_datasize = []
 
         for item in shapes:
+            if item == '':
+                self.input_shapes.append([])
+                continue
             elems = item.split(',')
             elems = [int(x) for x in elems]
             self.input_shapes.append(elems)
         for item in dtypes:
             elem = int(item)
             self.input_dtypes.append(elem)
         for i in range(len(shapes)):
-            elem_size = math.prod(self.input_shapes[i])
+            elem_size = math.prod(self.input_shapes[i]) if len(
+                self.input_shapes[i]) > 0 else 1
             self.input_datasize.append(
                 elem_size * acl.data_type_size(self.input_dtypes[i]))
         self.input_datasize_c = (

dicp/dicp/vendor/AscendGraph/compile_job.py

@@ -122,7 +122,6 @@ def __init__(self, source_code) -> None:
             str(self.device_id) + code_hash(compile_file_code)
         )
         self._output_graph_path = self._input_path[:-5] + '/graph'
-        # print('output_path: ', self._output_graph_path)
         self._model_path = [f'{self._output_graph_path}.om',
                             f'{self._output_graph_path}_linux_x86_64.om']
         self._lib_path = "/tmp/dicp_ascend/ge_graph.so"