未验证 提交 03479469 编写于 作者: 石晓伟 提交者: GitHub

fix multi-thread error of fc_gru_fuse_pass.cc, test=develop (#21841)

* fix multi-thread error of fc_gru_fuse_pass.cc, test=develop

* export FLAGS and GLOG symbols, test=develop
上级 4bb8df93
......@@ -92,8 +92,12 @@ static int BuildFusion(Graph* graph, const std::string& name_scope,
}
#undef GET_NODE
#define NEW_IMTERMEDIATE_OUT(key) \
scope.Var(NEW_NAME(key))->GetMutable<framework::LoDTensor>()
#define NEW_IMTERMEDIATE_OUT(key) \
VarDesc key(NEW_NAME(key)); \
key.SetPersistable(false); \
auto* key##_node = graph->CreateVarNode(&key); \
IR_NODE_LINK_TO(op, key##_node);
NEW_IMTERMEDIATE_OUT(ReorderedH0);
NEW_IMTERMEDIATE_OUT(XX);
NEW_IMTERMEDIATE_OUT(BatchedInput);
......
......@@ -31,7 +31,7 @@ DEFINE_string(
"'<space splitted floats as data>\t<space splitted ints as shape'");
DEFINE_bool(use_gpu, false, "Whether use gpu.");
#ifdef PADDLE_WITH_SHARED_LIB
DEFINE_bool(profile, false, "Whether use profile.");
DECLARE_bool(profile);
#endif
namespace paddle {
......
......@@ -2,7 +2,10 @@
global:
*paddle*;
*Pass*;
*profile*;
extern "C++" {
fL*::*;
};
local:
*;
};
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册