Browse Source

fix ut compile failed

tags/v1.3.0
lvliang chujinjin 5 years ago
parent
commit
116781041a
1 changed files with 5 additions and 5 deletions
  1. +5
    -5
      tests/ut/cpp/optimizer/ad/kpynative_test.cc

+ 5
- 5
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

Loading…
Cancel
Save