|
|
@@ -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 |