Skip to content

P
Paddle

BaiXuePrincess / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
提交 9db1a9e1 编写于 作者: Q Qiao Longfei
浏览文件
操作

change log level test=develop

上级 8342f12e
隐藏空白更改
内联 并排
Showing with 2 addition and 2 deletion
paddle/fluid/framework/details/multi_devices_graph_pass.h
浏览文件 @ 9db1a9e1
...@@ -127,12 +127,12 @@ class AsyncSSAGraphBuilder : public MultiDevSSAGraphBuilderBase { ...@@ -127,12 +127,12 @@ class AsyncSSAGraphBuilder : public MultiDevSSAGraphBuilderBase {
bool DealWithSpecialOp(ir::Graph *result, ir::Node *node) const override { bool DealWithSpecialOp(ir::Graph *result, ir::Node *node) const override {
if (node->Op()->Type() == "recv") { if (node->Op()->Type() == "recv") {
VLOG(0) << "set recv op do_not_run to true"; VLOG(1) << "set recv op do_not_run to true";
node->Op()->SetAttr("do_not_run", true); node->Op()->SetAttr("do_not_run", true);
node->Op()->Flush(); node->Op()->Flush();
} else if (node->Name() == "lookup_table" || node->Name() == "nce" || } else if (node->Name() == "lookup_table" || node->Name() == "nce" ||
node->Name() == "hierarchical_sigmoid") { node->Name() == "hierarchical_sigmoid") {
VLOG(0) << "set " << node->Name() << " op remote_prefetch to false"; VLOG(1) << "set " << node->Name() << " op remote_prefetch to false";
node->Op()->SetAttr("remote_prefetch", false); node->Op()->SetAttr("remote_prefetch", false);
} }
return false; return false;
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或