From 0a460f67985b2d8a22c57e18e57d4504a0533e01 Mon Sep 17 00:00:00 2001
From: zhangyanhui <zhangyanhui17@huawei.com>
Date: Wed, 28 Aug 2024 11:36:29 +0800
Subject: [PATCH] bugfix ofr fuzz problem

---
 mindspore/lite/src/litert/scheduler.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mindspore/lite/src/litert/scheduler.cc b/mindspore/lite/src/litert/scheduler.cc
index bc2cf881..01057ca9 100644
--- a/mindspore/lite/src/litert/scheduler.cc
+++ b/mindspore/lite/src/litert/scheduler.cc
@@ -1063,7 +1063,7 @@ int Scheduler::FindCpuKernel(const std::vector<Tensor *> &in_tensors, const std:
     MS_LOG(DEBUG) << "Get TypeId(expect = " << kernel_data_type << ", real = " << cpu_desc.data_type
                   << ") op success: " << PrimitiveCurVersionTypeName(op_type);
     if (is_train_session_) {
-      (*kernel)->Prepare();
+      ret = (*kernel)->Prepare();
       RestoreTensorData(&restored_origin_tensors);
     }
   }
-- 
2.25.1