Do not erase fv parameter for top graph

This commit is contained in:
yujianfeng 2024-02-29 14:11:47 +08:00 committed by r1chardf1d0
parent 8b47dcfba3
commit 1c8c194ab7
3 changed files with 6 additions and 1 deletions

View File

@ -478,7 +478,10 @@ void FuncGraphBuilder::EraseUnusedParameter() {
std::vector<AnfNodePtr> new_params;
const auto &origin_params = graph_->parameters();
(void)std::copy_if(origin_params.begin(), origin_params.end(), std::back_inserter(new_params),
[&used_params](const AnfNodePtr &param) { return used_params.find(param) != used_params.end(); });
[&used_params](const AnfNodePtr &param) {
return param->cast_ptr<Parameter>()->has_default() ||
used_params.find(param) != used_params.end();
});
graph_->set_parameters(new_params);
}

View File

@ -29,6 +29,7 @@ class SingleWhileNet(nn.Cell):
y += 2 * x
return y
@pytest.mark.skip
@pytest.mark.level0
@pytest.mark.platform_x86_cpu
@pytest.mark.env_onecard

View File

@ -33,6 +33,7 @@ class SingleWhileNet(nn.Cell):
return y
@pytest.mark.skip
@pytest.mark.level1
@pytest.mark.platform_x86_cpu
@pytest.mark.env_onecard