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
release/1.7
石晓伟 5 years ago committed by GitHub
parent 4bb8df9363
commit 03479469a7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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:
*;
};

Loading…
Cancel
Save