pull/820/head
wangxiaotian22 4 years ago
parent 0074b0b48f
commit 610828561c

@ -282,4 +282,42 @@ TEST_F(UtestDavinciModel, init_unknown) {
const vector<void *> outputs = { &virtual_addr };
EXPECT_EQ(model.UpdateKnownNodeArgs(inputs, outputs), SUCCESS);
}
TEST_F(UtestDavinciModel, ReturnNoOutput_test) {
DavinciModel model(0, nullptr);
GeTensorDesc tensor(GeShape(), FORMAT_NCHW, DT_FLOAT);
TensorUtils::SetSize(tensor, 512);
OpDescPtr var1 = CreateOpDesc("var1", VARIABLE);
var1->AddInputDesc(tensor);
var1->AddOutputDesc(tensor);
var1->SetInputOffset({1024});
var1->SetOutputOffset({1024});
model.variable_op_list_.push_back(var1);
EXPECT_EQ(model.ReturnNoOutput(model), PARAM_INVALID);
}
TEST_F(UtestDavinciModel, SyncVarData_test) {
DavinciModel model(0, nullptr);
GeTensorDesc tensor(GeShape(), FORMAT_NCHW, DT_FLOAT);
TensorUtils::SetSize(tensor, 512);
OpDescPtr var1 = CreateOpDesc("var1", VARIABLE);
var1->AddInputDesc(tensor);
var1->AddOutputDesc(tensor);
var1->SetInputOffset({1024});
var1->SetOutputOffset({1024});
model.variable_op_list_.push_back(var1);
EXPECT_NE(model.SyncVarData(model), SUCCESS);
}
} // namespace ge

Loading…
Cancel
Save