diff --git a/tests/ut/ge/graph/load/new_model_manager_model_manager_unittest.cc b/tests/ut/ge/graph/load/new_model_manager_model_manager_unittest.cc index 1c6e5a10..4096afa5 100644 --- a/tests/ut/ge/graph/load/new_model_manager_model_manager_unittest.cc +++ b/tests/ut/ge/graph/load/new_model_manager_model_manager_unittest.cc @@ -348,5 +348,18 @@ TEST_F(UtestModelManagerModelManager, destroy_aicpu_session) { manager.sess_ids_.insert(0); manager.DestroyAicpuSession(0); } +// test DataInputTensor +TEST_F(UtestModelManagerModelManager, test_data_input_tensor) { + ModelManager mm; + mm.model_map_={{1,make_shared()}}; + mm.hybrid_model_map_={{1,make_shared()}}; + ge::ModelData data; + data.model_len = 10; + uint32_t model_id = 1; + InputTensorInfo input_tensor = new InputTensorInfo(); + std::vector inputs = {input_tensor}; + auto ret = mm.DataInputTensor(1,inputs); + EXPECT_EQ(ge::SUCCESS, ret); +} } // namespace ge