-
Notifications
You must be signed in to change notification settings - Fork 1
/
arithematic.hpp
79 lines (56 loc) · 2.44 KB
/
arithematic.hpp
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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
#ifndef ARITHEMATIC_H
#define ARITHEMATIC_H
#include <iostream>
#include <array>
#include <algorithm>
#include <memory>
#include <cstdarg>
#include<string.h>
#include "ndarray.hpp"
#include "slice.hpp"
using namespace std;
template<typename T, size_t num_dimensions>
NdarrayMetadata<T, num_dimensions> add(NdarrayMetadata<T, num_dimensions> array1,NdarrayMetadata<T, num_dimensions> array2){
array<size_t, num_dimensions> out_shape = array1.shape;
vector<int64_t> output_allocated_memory(calc_size<size_t, num_dimensions>(out_shape));
auto output = create_array<int64_t, num_dimensions>(output_allocated_memory.data(), out_shape);
auto ite_output = output.data_buffer.get();
auto ite_array1 = array1.data_buffer.get();
auto ite_array2 = array2.data_buffer.get();
for(unsigned int i = 0; i< array1.total_size; i++){
ite_output[i] = ite_array1[i] + ite_array2[i];
cout << ite_output[i] << " " << i << endl;
}
return output;
}
template<typename T, size_t num_dimensions>
NdarrayMetadata<T, num_dimensions> subtract(NdarrayMetadata<T, num_dimensions> array1,NdarrayMetadata<T, num_dimensions> array2){
array<size_t, num_dimensions> out_shape = array1.shape;
vector<int64_t> output_allocated_memory(calc_size<size_t, num_dimensions>(out_shape));
auto output = create_array<int64_t, num_dimensions>(output_allocated_memory.data(), out_shape);
auto ite_output = output.data_buffer.get();
auto ite_array1 = array1.data_buffer.get();
auto ite_array2 = array2.data_buffer.get();
// auto output = array1;
for(unsigned int i = 0; i< array1.total_size; i++){
ite_output[i] = ite_array1[i] - ite_array2[i];
cout << ite_output[i] << " " << i << endl;
}
return output;
}
template<typename T, size_t num_dimensions>
NdarrayMetadata<T, num_dimensions> multiplie(NdarrayMetadata<T, num_dimensions> array1,NdarrayMetadata<T, num_dimensions> array2){
array<size_t, num_dimensions> out_shape = array1.shape;
vector<int64_t> output_allocated_memory(calc_size<size_t, num_dimensions>(out_shape));
auto output = create_array<int64_t, num_dimensions>(output_allocated_memory.data(), out_shape);
auto ite_output = output.data_buffer.get();
auto ite_array1 = array1.data_buffer.get();
auto ite_array2 = array2.data_buffer.get();
// auto output = array1;
for(unsigned int i = 0; i< array1.total_size; i++){
ite_output[i] = ite_array1[i] * ite_array2[i];
cout << ite_output[i] << " " << i << endl;
}
return output;
}
#endif