提交 72587055 编写于 作者: L LeGend-AI

rename ProcessMsgAndWardKernel to ProcessMsgWithKernelCtx

上级 6c2f2fba
......@@ -8,7 +8,7 @@ void CopyActor::Init(const TaskProto& task_proto) {
Actor::Init(task_proto);
}
void CopyActor::ProcessMsgAndWardKernel(const ActorMsg& msg,
void CopyActor::ProcessMsgWithKernelCtx(const ActorMsg& msg,
const KernelContext& kernel_ctx) {
if (TryOneReadDone(msg.regst_warpper()->regst_raw_ptr()) != 0) {
waiting_in_regst_.push(std::move(msg.regst_warpper()));
......
......@@ -15,7 +15,7 @@ public:
protected:
CopyActor() = default;
void ProcessMsgAndWardKernel(const ActorMsg& msg, const KernelContext& kernel_ctx);
void ProcessMsgWithKernelCtx(const ActorMsg& msg, const KernelContext& kernel_ctx);
private:
std::queue<std::shared_ptr<RegstWarpper>> waiting_in_regst_;
......
......@@ -7,7 +7,7 @@ namespace oneflow {
void CopyCommNetActor::ProcessMsg(const ActorMsg& msg,
const ThreadContext&) {
KernelContext kernel_ctx;
ProcessMsgAndWardKernel(msg, kernel_ctx);
ProcessMsgWithKernelCtx(msg, kernel_ctx);
}
REGISTER_ACTOR(kCopyCommNetTask, true, CopyCommNetActor);
......
......@@ -8,7 +8,7 @@ void CopyHdActor::ProcessMsg(const ActorMsg& msg,
const ThreadContext& thread_ctx) {
KernelContext kernel_ctx;
kernel_ctx.cuda_stream = thread_ctx.copy_hd_cuda_stream;
ProcessMsgAndWardKernel(msg, kernel_ctx);
ProcessMsgWithKernelCtx(msg, kernel_ctx);
}
REGISTER_ACTOR(kCopyHdTask, true, CopyHdActor);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册