blob: f97c79886a6906b165522a7ad9fee6f36396d3a4 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
// Copyright (C) 2018-2023 Intel Corporation
// SPDX-License-Identifier: Apache-2.0
//
#pragma once
#include <algorithm>
#include "openvino/runtime/allocator.hpp"
// Modified from SharedTensorAllocator in [openvinotoolkit/openvino ›
// samples/cpp/benchmark_app/shared_tensor_allocator.hpp]
class SharedTensorAllocator : public ov::AllocatorImpl {
public:
SharedTensorAllocator(size_t sizeBytes) : size(sizeBytes) { data = new char[size]; }
// Copy Constructor
SharedTensorAllocator(const SharedTensorAllocator& other) : size(other.size) {
data = new char[size];
std::copy(other.data, other.data + size, data);
}
// Copy Assignment Operator
SharedTensorAllocator& operator=(const SharedTensorAllocator& other) {
if (this != &other) {
size = other.size;
delete[] data;
data = new char[size];
std::copy(other.data, other.data + size, data);
}
return *this;
}
~SharedTensorAllocator() { delete[] data; }
void* allocate(const size_t bytes, const size_t) override {
return bytes <= this->size ? (void*)data : nullptr;
}
void deallocate(void* handle, const size_t bytes, const size_t) override {
if (handle == data) {
delete[] data;
data = nullptr;
}
}
bool is_equal(const AllocatorImpl& other) const override {
auto other_tensor_allocator = dynamic_cast<const SharedTensorAllocator*>(&other);
return other_tensor_allocator != nullptr && other_tensor_allocator == this;
}
char* get_buffer() { return data; }
private:
char* data;
size_t size;
};
|
