Skip to content

Commit 60ecbc4

Browse files
authored
Fix usage of unique_ptr for arrays in data_objects_test (#5837)
Replace `unique_ptr<element_t>` with `unique_ptr<element_t[]>`, because they hold arrays allocated with `operator new[]`. Signed-off-by: Michal Zientkiewicz <[email protected]>
1 parent c17f33c commit 60ecbc4

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

dali/c_api_2/data_objects_test.cc

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -184,7 +184,7 @@ TEST(CAPI2_TensorListTest, AttachBuffer) {
184184
{ 1080, 1920, 3 }
185185
});
186186
auto size = lshape.num_elements();
187-
std::unique_ptr<element_t> data(new element_t[size]);
187+
std::unique_ptr<element_t[]> data(new element_t[size]);
188188

189189
ptrdiff_t offsets[4] = {};
190190
for (int i = 1; i < 4; i++)
@@ -242,7 +242,7 @@ TEST(CAPI2_TensorListTest, AttachSamples) {
242242
});
243243
auto size = lshape.num_elements();
244244
int N = lshape.num_samples();
245-
std::vector<std::unique_ptr<element_t>> data(N);
245+
std::vector<std::unique_ptr<element_t[]>> data(N);
246246

247247
for (int i = 0; i < N; i++) {
248248
data[i].reset(new element_t[size]);
@@ -310,7 +310,7 @@ TEST(CAPI2_TensorListTest, ViewAsTensor) {
310310
daliDataType_t dtype = dali::type2id<element_t>::value;
311311
dali::TensorListShape<> lshape = dali::uniform_list_shape(4, { 480, 640, 3 });
312312
auto size = lshape.num_elements();
313-
std::unique_ptr<element_t> data(new element_t[size]);
313+
std::unique_ptr<element_t[]> data(new element_t[size]);
314314

315315
ptrdiff_t sample_size = volume(lshape[0]) * sizeof(element_t);
316316

@@ -385,7 +385,7 @@ TEST(CAPI2_TensorListTest, ViewAsTensorError) {
385385
daliDataType_t dtype = dali::type2id<element_t>::value;
386386
dali::TensorListShape<> lshape = dali::uniform_list_shape(4, { 480, 640, 3 });
387387
auto size = lshape.num_elements();
388-
std::unique_ptr<element_t> data(new element_t[size]);
388+
std::unique_ptr<element_t[]> data(new element_t[size]);
389389

390390
ptrdiff_t sample_size = volume(lshape[0]) * sizeof(element_t);
391391

0 commit comments

Comments
 (0)