Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MegEngine 天元
MegEngine
提交
93f4977c
MegEngine
项目概览
MegEngine 天元
/
MegEngine
1 年多 前同步成功
通知
404
Star
4705
Fork
582
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
MegEngine
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
提交
93f4977c
编写于
12月 04, 2020
作者:
M
Megvii Engine Team
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
feat(mge/imperative): add thread name
GitOrigin-RevId: 61eece55b8f4008882532fa4b50a94f1164efee4
上级
98a74e4a
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
13 addition
and
1 deletion
+13
-1
imperative/python/src/helper.h
imperative/python/src/helper.h
+4
-0
imperative/src/impl/interpreter_impl.h
imperative/src/impl/interpreter_impl.h
+6
-1
imperative/src/impl/physical_tensor.cpp
imperative/src/impl/physical_tensor.cpp
+3
-0
未找到文件。
imperative/python/src/helper.h
浏览文件 @
93f4977c
...
...
@@ -69,6 +69,10 @@ class PyTaskDipatcher {
pybind11
::
gil_scoped_acquire
_
;
f
();
}
void
on_async_queue_worker_thread_start
()
override
{
mgb
::
sys
::
set_thread_name
(
"py_task_worker"
);
}
};
Queue
queue
;
bool
finalized
=
false
;
...
...
imperative/src/impl/interpreter_impl.h
浏览文件 @
93f4977c
...
...
@@ -143,10 +143,15 @@ private:
size_t
m_enable_evict
=
0
;
struct
WorkQueue
:
AsyncQueueSC
<
Command
,
WorkQueue
>
{
WorkQueue
(
ChannelImpl
*
owner
)
:
m_owner
(
owner
)
{}
WorkQueue
(
ChannelImpl
*
owner
)
:
m_owner
(
owner
)
{
sys
::
set_thread_name
(
"interpreter"
);
}
void
process_one_task
(
Command
&
cmd
)
{
m_owner
->
process_one_task
(
cmd
);
}
void
on_async_queue_worker_thread_start
()
override
{
sys
::
set_thread_name
(
"worker"
);
}
private:
ChannelImpl
*
m_owner
;
}
m_worker
;
...
...
imperative/src/impl/physical_tensor.cpp
浏览文件 @
93f4977c
...
...
@@ -44,6 +44,9 @@ class AsyncReleaser : public CompNodeDepedentObject {
std
::
this_thread
::
sleep_for
(
1us
);
add_task
(
std
::
move
(
param
));
}
void
on_async_queue_worker_thread_start
()
override
{
sys
::
set_thread_name
(
"releaser"
);
}
};
Waiter
m_waiter
{
this
};
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录