blob: 08610b229d70d572ac4379e1179222720ce7d1c9 (
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
|
#pragma once
#include <cstdlib>
// 64-byte aligned simple vector.
namespace intgemm {
template <class T> class AlignedVector {
public:
explicit AlignedVector(std::size_t size)
: mem_(static_cast<T*>(aligned_alloc(64, size * sizeof(T)))) {}
~AlignedVector() { std::free(mem_); }
T &operator[](std::size_t offset) { return mem_[offset]; }
const T &operator[](std::size_t offset) const { return mem_[offset]; }
T *get() { return mem_; }
const T *get() const { return mem_; }
private:
T *mem_;
// Deleted.
AlignedVector(AlignedVector &);
AlignedVector &operator=(AlignedVector &);
};
} // namespace intgemm
|