修复graphllm的内存泄漏

ztxz16 · Sep 26, 2024 · a6c6c43 · a6c6c43
1 parent 7502e4c
commit a6c6c43
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/graph.cpp b/src/graph.cpp
@@ -112,6 +112,7 @@ namespace fastllm {
         for (auto &node : graph.nodes) {
             if (allDatas.find(node.name) == allDatas.end()) {
                 allDatas[node.name] = new Data();
+                tempDatas[node.name] = allDatas[node.name];
             }
         }
         Data emptyData;