fix duration assertion and index bug (#1123)

Author: mengniwang95

neural_compressor/adaptor/ox_utils/onnxrt_mid.py

@@ -132,8 +132,8 @@ def augment_graph(self, activation_only=False, weight_only=False):
                         tensors_to_dump.update(node.input)
                     tensors_to_dump.update(node.output)
                     if node.op_type == 'EmbedLayerNormalization' and len(node.output) > 1 and \
-                        node.output[2] in tensors_to_dump:
-                        tensors_to_dump.remove(node.output[2])
+                        node.output[1] in tensors_to_dump:
+                        tensors_to_dump.remove(node.output[1])
                 elif weight_only:
                     for input in node.input:
                         if self.already_quantized and \

neural_compressor/objective.py

@@ -134,7 +134,7 @@ def start(self):
         self.start_time = time.time()
     def end(self):
         self.duration = time.time() - self.start_time
-        assert self.duration > 0, 'please use start() before end()'
+        assert self.duration >= 0, 'please use start() before end()'
         self._result_list.append(self.duration)
 
 @objective_registry