binary buffer by vector<uint8_t>

openvinotoolkit · Nov 15, 2022 · 0b6465f · 0b6465f
1 parent 424871b
commit 0b6465f
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/plugins/intel_gpu/src/graph/primitive_inst.cpp b/src/plugins/intel_gpu/src/graph/primitive_inst.cpp
@@ -1101,10 +1101,10 @@ void primitive_inst::load(cldnn::BinaryInputBuffer& ib) {
         if (_allocation_type == allocation_type::usm_host || _allocation_type == allocation_type::usm_shared) {
             ib >> cldnn::make_data(_outputs[0]->buffer_ptr(), data_size);
         } else {
-            char *_buf = new char[data_size];
-            ib >> cldnn::make_data(_buf, data_size);
-            _outputs[0]->copy_from(get_network().get_stream(), _buf);
-            delete[] _buf;
+            std::vector<uint8_t> _buf;
+            _buf.resize(data_size);
+            ib >> cldnn::make_data(_buf.data(), data_size);
+            _outputs[0]->copy_from(get_network().get_stream(), _buf.data());
         }
     } else if (_object_type == object_type::EXECUTABLE_INST) {
         _impl_params.release();