[fix] Fix inference issues caused by memory synchronization

Author: LittleMouse

projects/llm_framework/main_llm/src/runner/ax_model_runner/ax_model_runner_ax650.cpp

@@ -432,10 +432,23 @@ void ax_runner_ax650::deinit()
 
 int ax_runner_ax650::inference()
 {
-    return AX_ENGINE_RunSync(m_handle->handle, &m_handle->io_data[0]);
+    int ret = AX_ENGINE_RunSync(m_handle->handle, &m_handle->io_data[0]);
+    for (size_t i = 0; i < get_num_outputs(); i++)
+    {
+        auto &tensor = get_output(i);
+        AX_SYS_MinvalidateCache(tensor.phyAddr, tensor.pVirAddr, tensor.nSize);
+    }
+    return ret;
 }
 
 int ax_runner_ax650::inference(int grpid)
 {
-    return AX_ENGINE_RunGroupIOSync(m_handle->handle, m_handle->context, grpid, &m_handle->io_data[grpid]);
+    int ret = AX_ENGINE_RunGroupIOSync(m_handle->handle, m_handle->context, grpid, &m_handle->io_data[grpid]);
+
+    for (size_t i = 0; i < get_num_outputs(); i++)
+    {
+        auto &tensor = get_output(grpid, i);
+        AX_SYS_MinvalidateCache(tensor.phyAddr, tensor.pVirAddr, tensor.nSize);
+    }
+    return ret;
 }