提交 37f6b9ab 编写于 作者: Q Qiao Longfei

fix build test=develop

上级 30618409
...@@ -54,7 +54,7 @@ class MultiDevSSAGraphBuilderBase : public ir::Pass { ...@@ -54,7 +54,7 @@ class MultiDevSSAGraphBuilderBase : public ir::Pass {
bool UseGPU() const; bool UseGPU() const;
bool NeedCollectiveForGrad(const std::string &grad_name, virtual bool NeedCollectiveForGrad(const std::string &grad_name,
std::vector<ir::Node *> ops) const; std::vector<ir::Node *> ops) const;
bool IsScaleLossOp(ir::Node *node) const; bool IsScaleLossOp(ir::Node *node) const;
...@@ -117,7 +117,10 @@ class AsyncSSAGraphBuilder : public MultiDevSSAGraphBuilderBase { ...@@ -117,7 +117,10 @@ class AsyncSSAGraphBuilder : public MultiDevSSAGraphBuilderBase {
void InsertCollectiveOp(ir::Graph *result, const std::string &p_name, void InsertCollectiveOp(ir::Graph *result, const std::string &p_name,
const std::string &g_name) const override {} const std::string &g_name) const override {}
bool NeedCollectiveOps() const override { return false; } bool NeedCollectiveForGrad(const std::string &grad_name,
std::vector<ir::Node *> ops) const {
return false;
}
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") {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册