Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Oneflow-Inc
oneflow
提交
3f12dd8f
O
oneflow
项目概览
Oneflow-Inc
/
oneflow
上一次同步 2 年多
通知
13
Star
2733
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
O
oneflow
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
未验证
提交
3f12dd8f
编写于
1月 26, 2021
作者:
J
Juncheng
提交者:
GitHub
1月 26, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Remove useless logical2sorted_in/out_box (#4152)
上级
92fe76f8
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
2 addition
and
6 deletion
+2
-6
oneflow/core/graph/logical_node.h
oneflow/core/graph/logical_node.h
+0
-2
oneflow/core/graph/task_graph.cpp
oneflow/core/graph/task_graph.cpp
+2
-4
未找到文件。
oneflow/core/graph/logical_node.h
浏览文件 @
3f12dd8f
...
@@ -93,8 +93,6 @@ class LogicalNode : public Node<LogicalNode, LogicalEdge> {
...
@@ -93,8 +93,6 @@ class LogicalNode : public Node<LogicalNode, LogicalEdge> {
(const LogicalNode* src_logical, const LogicalNode* dst_logical, \
(const LogicalNode* src_logical, const LogicalNode* dst_logical, \
const std::vector<CompTaskNode*>& sorted_src_comp_tasks, \
const std::vector<CompTaskNode*>& sorted_src_comp_tasks, \
const std::vector<CompTaskNode*>& sorted_dst_comp_tasks, \
const std::vector<CompTaskNode*>& sorted_dst_comp_tasks, \
HashMap<const LogicalNode*, std::vector<TaskNode*>>* logical2sorted_in_box, \
HashMap<const LogicalNode*, std::vector<TaskNode*>>* logical2sorted_out_box, \
std::function<TaskNode**(CompTaskNode * src, int64_t machine_id, int32_t mem_zone_id)> \
std::function<TaskNode**(CompTaskNode * src, int64_t machine_id, int32_t mem_zone_id)> \
MutBufTask, \
MutBufTask, \
std::function<int64_t(const TaskNode*)> AllocateCpuThrdIdEvenly)
std::function<int64_t(const TaskNode*)> AllocateCpuThrdIdEvenly)
...
...
oneflow/core/graph/task_graph.cpp
浏览文件 @
3f12dd8f
...
@@ -228,8 +228,6 @@ TaskGraph::TaskGraph(std::unique_ptr<const LogicalGraph>&& logical_gph) {
...
@@ -228,8 +228,6 @@ TaskGraph::TaskGraph(std::unique_ptr<const LogicalGraph>&& logical_gph) {
builders
.
emplace_back
(
new
NaiveB2PSubTskGphBuilder
());
builders
.
emplace_back
(
new
NaiveB2PSubTskGphBuilder
());
sub_tsk_gph_builder_
.
reset
(
new
ChainSubTskGphBuilder
(
builders
));
sub_tsk_gph_builder_
.
reset
(
new
ChainSubTskGphBuilder
(
builders
));
HashMap
<
const
LogicalNode
*
,
std
::
vector
<
CompTaskNode
*>>
logical2sorted_comp_tasks
;
HashMap
<
const
LogicalNode
*
,
std
::
vector
<
CompTaskNode
*>>
logical2sorted_comp_tasks
;
HashMap
<
const
LogicalNode
*
,
std
::
vector
<
TaskNode
*>>
logical2sorted_in_box
;
HashMap
<
const
LogicalNode
*
,
std
::
vector
<
TaskNode
*>>
logical2sorted_out_box
;
HashMap
<
CompTaskNode
*
,
HashMap
<
int64_t
,
std
::
vector
<
TaskNode
*>>>
buf_task
;
HashMap
<
CompTaskNode
*
,
HashMap
<
int64_t
,
std
::
vector
<
TaskNode
*>>>
buf_task
;
auto
MutBufTask
=
[
&
](
CompTaskNode
*
task_node
,
int64_t
machine_id
,
int32_t
mem_zone_id
)
{
auto
MutBufTask
=
[
&
](
CompTaskNode
*
task_node
,
int64_t
machine_id
,
int32_t
mem_zone_id
)
{
auto
&
buf_vec
=
buf_task
[
task_node
][
machine_id
];
auto
&
buf_vec
=
buf_task
[
task_node
][
machine_id
];
...
@@ -265,8 +263,8 @@ TaskGraph::TaskGraph(std::unique_ptr<const LogicalGraph>&& logical_gph) {
...
@@ -265,8 +263,8 @@ TaskGraph::TaskGraph(std::unique_ptr<const LogicalGraph>&& logical_gph) {
GetMthdForBldSubTskGph
(
logical_edge
->
src_node
(),
logical_edge
->
dst_node
());
GetMthdForBldSubTskGph
(
logical_edge
->
src_node
(),
logical_edge
->
dst_node
());
(
this
->*
method
)(
logical_edge
->
src_node
(),
logical_edge
->
dst_node
(),
(
this
->*
method
)(
logical_edge
->
src_node
(),
logical_edge
->
dst_node
(),
logical2sorted_comp_tasks
.
at
(
logical_edge
->
src_node
()),
logical2sorted_comp_tasks
.
at
(
logical_edge
->
src_node
()),
logical2sorted_comp_tasks
.
at
(
logical_edge
->
dst_node
()),
&
logical2sorted_in_box
,
logical2sorted_comp_tasks
.
at
(
logical_edge
->
dst_node
()),
MutBufTask
,
&
logical2sorted_out_box
,
MutBufTask
,
AllocateCpuThrdIdEvenly
);
AllocateCpuThrdIdEvenly
);
SetAreaIdForNewNodes
(
logical_edge
->
src_node
(),
logical_edge
->
dst_node
());
SetAreaIdForNewNodes
(
logical_edge
->
src_node
(),
logical_edge
->
dst_node
());
});
});
logical_gph_
->
ForEachNecessaryCtrlEdge
(
logical_gph_
->
ForEachNecessaryCtrlEdge
(
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录