Fix accuracy issue

riverlijunjie · Sep 15, 2024 · 8a5489d · 8a5489d
1 parent 1202a12
commit 8a5489d
Show file tree

Hide file tree

Showing 4 changed files with 507 additions and 94 deletions.
diff --git a/src/plugins/intel_gpu/include/intel_gpu/plugin/sub_memory_manager.hpp b/src/plugins/intel_gpu/include/intel_gpu/plugin/sub_memory_manager.hpp
@@ -24,6 +24,7 @@ class SubMemoryManager {
         std::shared_ptr<cldnn::stream> stream_ptr;
         std::vector<cldnn::memory::ptr> recv_bufs;  // todo: avoid mem copy
         std::vector<cldnn::event::ptr> events;
+        void* sync_buf; // sync cl buf
     };
 
     SubMemoryManager(int num_sub_streams) {
@@ -32,7 +33,8 @@ class SubMemoryManager {
         MemoryInfo memory_info;
         memory_info.flag = false;
         memory_info.last_used = false;
-        memory_info.events.assign(2, nullptr);
+        memory_info.sync_buf = nullptr;
+        memory_info.events.assign(_num_sub_streams, nullptr);
         std::vector<MemoryInfo> memorys;
         memorys.assign(_num_sub_streams, memory_info);
         _memorys_table.assign(2, memorys);