apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2026-04-23 00:17:25 +08:00
Code Issues Actions 19 Packages Projects Releases Wiki Activity

[BugFix] replace ftok with custom_ftok in get_output/save_output ops (#6822)

Browse Source 
* [BugFix] replace ftok with custom_ftok in get_output/save_output ops

* [Test] add unit test for custom_ftok

* [Chore] create custom_ftok.h

* [Chore] reorganize header file

* [Fix] fix cache messager msg_queue_id+rank_id conflict
This commit is contained in:
Yonghua Li
2026-03-16 14:22:18 +08:00
committed by GitHub
parent 4d39232553
commit 7c8c0a3c02
24 changed files with 126 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files
custom_ops/gpu_ops/save_with_output_msg.cc
+2 -1
View File
@@ -13,6 +13,7 @@
// limitations under the License.
#include "save_with_output_msg.h"
#include "custom_ftok.h"
void save_kernel(const paddle::Tensor& x,
const paddle::Tensor& not_need_stop,
@@ -68,7 +69,7 @@ void save_kernel(const paddle::Tensor& x,
#ifdef SAVE_WITH_OUTPUT_DEBUG
std::cout << "msg_queue_id is: " << msg_queue_id << std::endl;
#endif
static key_t key = ftok("/dev/shm", msg_queue_id);
static key_t key = custom_ftok("/dev/shm", msg_queue_id);
static int msgid = msgget(key, IPC_CREAT | 0666);
#ifdef SAVE_WITH_OUTPUT_DEBUG