From 116781041ad996b0baa5b9eef58972004a19a85f Mon Sep 17 00:00:00 2001 From: lvliang Date: Tue, 11 May 2021 09:49:41 +0800 Subject: [PATCH] fix ut compile failed --- tests/ut/cpp/optimizer/ad/kpynative_test.cc | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/tests/ut/cpp/optimizer/ad/kpynative_test.cc b/tests/ut/cpp/optimizer/ad/kpynative_test.cc index 2c141387762..2ffb0ef5dea 100644 --- a/tests/ut/cpp/optimizer/ad/kpynative_test.cc +++ b/tests/ut/cpp/optimizer/ad/kpynative_test.cc @@ -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