File tree Expand file tree Collapse file tree 1 file changed +3
-2
lines changed
yql/essentials/minikql/computation Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -37,8 +37,9 @@ class TBlockTrimmerBase : public IBlockTrimmer {
37
37
return result;
38
38
}
39
39
40
+ template <typename TBuffer = NUdf::TResizeableBuffer>
40
41
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_);
42
43
ARROW_OK (buffer->Resize (size, false ));
43
44
return buffer;
44
45
}
@@ -92,7 +93,7 @@ class TResourceBlockTrimmer : public TBlockTrimmerBase {
92
93
auto origData = array->GetValues <NUdf::TUnboxedValue>(1 );
93
94
auto dataSize = sizeof (NUdf::TUnboxedValue) * array->length ;
94
95
95
- auto trimmedBuffer = CreateResizableBuffer (dataSize);
96
+ auto trimmedBuffer = CreateResizableBuffer<NUdf::TResizableManagedBuffer<NUdf::TUnboxedValue>> (dataSize);
96
97
auto trimmedBufferData = reinterpret_cast <NUdf::TUnboxedValue*>(trimmedBuffer->mutable_data ());
97
98
98
99
for (int64_t i = 0 ; i < array->length ; i++) {
You can’t perform that action at this time.
0 commit comments