saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-04 23:59:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix a memory leak in our testing code (#5312)

Browse source
This commit is contained in:
Changming Sun 2020-09-28 16:00:57 -07:00 committed by GitHub
parent 3693f91218
commit d45d68fdd4
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
onnxruntime/test/shared_lib/test_nontensor_types.cc
View file

@ -179,11 +179,13 @@ TEST(CApiTest, TypeInfoMap) {
#if !defined(DISABLE_ML_OPS)
Ort::Value map_ort = Ort::Value::CreateMap(keys_tensor, values_tensor);
Ort::TypeInfo type_info = map_ort.GetTypeInfo();
//It doesn't own the pointer
Ort::MapTypeInfo map_type_info = type_info.GetMapTypeInfo();
//Check key type
ASSERT_EQ(map_type_info.GetMapKeyType(), ONNX_TENSOR_ELEMENT_DATA_TYPE_INT64);
//It owns the pointer
Ort::TypeInfo map_value_type_info = map_type_info.GetMapValueType();
//Check value type and shape
@ -192,7 +194,7 @@ TEST(CApiTest, TypeInfoMap) {
// ASSERT_EQ(map_value_type_info.GetTensorTypeAndShapeInfo().GetShape(), dims);
ASSERT_EQ(map_value_type_info.GetTensorTypeAndShapeInfo().GetElementType(), ONNX_TENSOR_ELEMENT_DATA_TYPE_FLOAT);
map_value_type_info.release();
//As it doesn't own the pointer, we should release the ownership, otherwise the pointer will be double freed.
map_type_info.release();
#else