|
1 | 1 | #include <iostream> |
2 | 2 | #include <cstdlib> |
| 3 | +#include <vector> |
3 | 4 |
|
4 | 5 | #include "my_alloc.h" |
5 | 6 |
|
6 | 7 | int main(){ |
7 | 8 |
|
8 | | - size_t d1[1] = {1}; |
9 | | - size_t d2[2] = {1,1}; |
10 | | - size_t d3[3] = {1,1,1}; |
11 | | - size_t d4[4] = {1,1,1,1}; |
| 9 | + const std::vector<size_t> d1 = {1}; |
| 10 | + const std::vector<size_t> d2 = {1,1}; |
| 11 | + const std::vector<size_t> d3 = {1,1,1}; |
| 12 | + const std::vector<size_t> d4 = {1,1,1,1}; |
12 | 13 |
|
13 | 14 | float *A1, *B1, *A2, *B2, *A3, *B3, *A4, *B4; |
14 | 15 |
|
15 | 16 |
|
16 | | - alloc1(&A1, &B1, d1); |
| 17 | + alloc1(&A1, &B1, &d1.front()); |
17 | 18 | std::cout << "1D: A and B allocated\n"; |
18 | 19 |
|
19 | | - dealloc1(&A1, &B1, d1); |
| 20 | + dealloc1(&A1, &B1, &d1.front()); |
20 | 21 | std::cout << "1D: A and B deallocated\n"; |
21 | 22 |
|
22 | | - alloc2(&A2, &B2, d2); |
| 23 | + alloc2(&A2, &B2, &d2.front()); |
23 | 24 | std::cout << "2D: A and B allocated\n"; |
24 | 25 |
|
25 | | - dealloc2(&A2, &B2, d2); |
| 26 | + dealloc2(&A2, &B2, &d2.front()); |
26 | 27 | std::cout << "2D: A and B deallocated\n"; |
27 | 28 |
|
28 | | - alloc3(&A3, &B3, d3); |
| 29 | + alloc3(&A3, &B3, &d3.front()); |
29 | 30 | std::cout << "3D: A and B allocated\n"; |
30 | 31 |
|
31 | | - dealloc3(&A3, &B3, d3); |
| 32 | + dealloc3(&A3, &B3, &d3.front()); |
32 | 33 | std::cout << "3D: A and B deallocated\n"; |
33 | 34 |
|
34 | | - alloc4(&A4, &B4, d4); |
| 35 | + alloc4(&A4, &B4, &d4.front()); |
35 | 36 | std::cout << "4D: A and B allocated\n"; |
36 | 37 |
|
37 | | - dealloc4(&A4, &B4, d4); |
| 38 | + dealloc4(&A4, &B4, &d4.front()); |
38 | 39 | std::cout << "4D: A and B deallocated\n"; |
39 | 40 |
|
40 | 41 | std::cout << "OK: allocate test\n"; |
|
0 commit comments