Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix ut compile failed

This commit is contained in:
lvliang 2021-05-11 09:49:41 +08:00 committed by chujinjin
parent 5c8130821c
commit 116781041a
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/ut/cpp/optimizer/ad/kpynative_test.cc
View File

@ -98,7 +98,7 @@ class TestKPynative : public UT::Common {
GradPynativeOp(k_pynative_cell, c_node, args, out);
}
}
auto bprop_fg = GradPynativeCellBuildFormalBProp(k_pynative_cell, AnfNodePtrList{}, true, false);
auto bprop_fg = GradPynativeCellEnd(k_pynative_cell, AnfNodePtrList{}, true, false, false, true);
return bprop_fg;
}
};
@ -106,23 +106,23 @@ class TestKPynative : public UT::Common {
TEST_F(TestKPynative, test_simple_add) {
auto primal_fg = BuildPrimalFuncGraph("test_simple_add");
resource->manager()->KeepRoots({primal_fg});
ExportIR(primal_fg->ToString() + ".dat", "", primal_fg);
ExportIR(primal_fg->ToString() + ".dat", primal_fg);
auto bprop_fg = BuildBpropFuncGraph(primal_fg);
resource->manager()->KeepRoots({bprop_fg});
ExportIR(bprop_fg->ToString() + ".dat", "", bprop_fg);
ExportIR(bprop_fg->ToString() + ".dat", bprop_fg);
}
TEST_F(TestKPynative, test_stop_gradient) {
auto primal_fg = BuildStopGradient("test_stop_gradient");
resource->manager()->KeepRoots({primal_fg});
ExportIR(primal_fg->ToString() + ".dat", "", primal_fg);
ExportIR(primal_fg->ToString() + ".dat", primal_fg);
auto bprop_fg = BuildBpropFuncGraph(primal_fg);
resource->manager()->KeepRoots({bprop_fg});
ExportIR(bprop_fg->ToString() + ".dat", "", bprop_fg);
ExportIR(bprop_fg->ToString() + ".dat", bprop_fg);
}
} // namespace ad
} // namespace mindspore