Skip to content

Commit 583d1db

Browse files
committed
Fixed trimmer
1 parent 26eb8a8 commit 583d1db

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

yql/essentials/minikql/computation/mkql_block_trimmer.cpp

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,8 +37,9 @@ class TBlockTrimmerBase : public IBlockTrimmer {
3737
return result;
3838
}
3939

40+
template<typename TBuffer = NUdf::TResizeableBuffer>
4041
std::unique_ptr<arrow::ResizableBuffer> CreateResizableBuffer(size_t size) const {
41-
auto buffer = NUdf::AllocateResizableBuffer(size, Pool_);
42+
auto buffer = NUdf::AllocateResizableBuffer<TBuffer>(size, Pool_);
4243
ARROW_OK(buffer->Resize(size, false));
4344
return buffer;
4445
}
@@ -92,7 +93,7 @@ class TResourceBlockTrimmer : public TBlockTrimmerBase {
9293
auto origData = array->GetValues<NUdf::TUnboxedValue>(1);
9394
auto dataSize = sizeof(NUdf::TUnboxedValue) * array->length;
9495

95-
auto trimmedBuffer = CreateResizableBuffer(dataSize);
96+
auto trimmedBuffer = CreateResizableBuffer<NUdf::TResizableManagedBuffer<NUdf::TUnboxedValue>>(dataSize);
9697
auto trimmedBufferData = reinterpret_cast<NUdf::TUnboxedValue*>(trimmedBuffer->mutable_data());
9798

9899
for (int64_t i = 0; i < array->length; i++) {

0 commit comments

Comments
 (0)