mooncake-track
/
Mooncake
0
0
Fork 12
Code Issues Pull Requests 13 Actions Packages Projects Releases Wiki Activity

[CCF Archive] Store object type eviction policy submission #3

Closed
kancel wants to merge 382 commits from kancel:ccf-archive-pr2746 into main
pull from: kancel:ccf-archive-pr2746
merge into: mooncake-track:main
Conversation 0 Commits 382 Files Changed 654 +23 -3
2 changed files with 23 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 59f62c3686 - Show all commits

12
mooncake-transfer-engine/src/transport/efa_transport/efa_transport.cpp
View File

@ -779,7 +779,17 @@ Status EfaTransport::submitTransfer(
size_t task_id = batch_desc.task_list.size();
batch_desc.task_list.resize(task_id + entries.size());
std::vector<TransferTask*> task_list;
for (auto& task : batch_desc.task_list) task_list.push_back(&task);
for (auto& request : entries) {
auto& task = batch_desc.task_list[task_id];
++task_id;
task.batch_id = batch_id;
#ifdef USE_ASCEND_HETEROGENEOUS
task.request = const_cast<TransferRequest*>(&request);
#else
task.request = &request;
#endif
task_list.push_back(&task);
}
return submitTransferTask(task_list);
}

14
mooncake-transfer-engine/src/transport/kunpeng_transport/ub_transport.cpp
View File

@ -186,8 +186,18 @@ Status UbTransport::submitTransfer(
size_t task_id = batch_desc.task_list.size();
batch_desc.task_list.resize(task_id + entries.size());
std::vector<TransferTask*> task_list;
task_list.reserve(batch_desc.task_list.size());
for (auto& task : batch_desc.task_list) task_list.push_back(&task);
task_list.reserve(entries.size());
for (auto& request : entries) {
auto& task = batch_desc.task_list[task_id];
++task_id;
task.batch_id = batch_id;
#ifdef USE_ASCEND_HETEROGENEOUS
task.request = const_cast<TransferRequest*>(&request);
#else
task.request = &request;
#endif
task_list.push_back(&task);
}
return submitTransferTask(task_list);
}