(bugfix): free cast_op if not null

ce7548d9 · zhaoying · jackzhang235 · eb1d5c50 · ce7548d9
隐藏空白更改
内联并排

Showing with 3 addition and 0 deletion

lite/kernels/mlu/bridges/argmax_op.cc lite/kernels/mlu/bridges/argmax_op.cc +3 -0

未找到文件。
--- a/lite/kernels/mlu/bridges/argmax_op.cc
+++ b/lite/kernels/mlu/bridges/argmax_op.cc
@@ -91,6 +91,9 @@ int ArgmaxConverter(void* ctx, OpLite* op, KernelBase* kernel) {
  }
  graph->FuseOp(argmax_op);
  CNML_CALL(cnmlDestroyBaseOp(&argmax_op));
+  if (cast_op) {
+    CNML_CALL(cnmlDestroyBaseOp(&cast_op));
+  }
  return SUCCESS;
 }