forked from lcompilers/lpython
-
Notifications
You must be signed in to change notification settings - Fork 0
/
elemental_05.py
131 lines (97 loc) · 4.07 KB
/
elemental_05.py
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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
from lpython import i32, f64, f32
from numpy import empty, sinh, cosh, reshape, int32, float32, float64, sin
def verify1d(array: f32[:], result: f32[:], size: i32):
i: i32
eps: f32 = f32(1e-6)
for i in range(size):
assert abs(sinh(sinh(array[i])) - result[i]) <= eps
def verifynd(array: f64[:, :, :, :], result: f64[:, :, :, :], size1: i32, size2: i32, size3: i32, size4: i32):
i: i32; size: i32;
eps: f64 = 1e-12
shape: i32[1] = empty((1,), dtype=int32)
size = size1 * size2 * size3 * size4
shape[0] = size
array1d: f64[12800] = reshape(array, shape)
result1d: f64[12800] = reshape(result, shape)
for i in range(size):
assert abs((sinh(array1d[i]) + 2.0)/2.0 - result1d[i]) <= eps
def elemental_sinh():
i: i32; j: i32; k: i32; l: i32; size: i32;
array1d: f32[10] = empty(10, dtype=float32)
sinh1d: f32[10] = empty(10, dtype=float32)
for i in range(10):
array1d[i] = f32(f64(i)/10.0)
sinh1d = sinh(sinh(array1d))
verify1d(array1d, sinh1d, 10)
arraynd: f64[40, 10, 16, 2] = empty((40, 10, 16, 2), dtype=float64)
sinhnd: f64[40, 10, 16, 2] = empty((40, 10, 16, 2), dtype=float64)
size = 40 * 10 * 16 * 2
for i in range(40):
for j in range(10):
for k in range(16):
for l in range(2):
arraynd[i, j, k, l] = float(i + 2*j + 3*k + 4*k)/float(size)
sinhnd = (sinh(arraynd) + 2.0)/2.0
verifynd(arraynd, sinhnd, 40, 10, 16, 2)
def verify2d(array: f64[:, :], result: f64[:, :], size1: i32, size2: i32):
i: i32; j: i32;
eps: f64 = 1e-12
for i in range(size1):
for j in range(size2):
assert abs(cosh(5.0 + array[i, j])**2.0 - result[i, j]) <= eps
def elemental_cosh():
i: i32; j: i32
array2d: f64[20, 10] = empty((20, 10), dtype=float64)
cosh2d: f64[20, 10] = empty((20, 10), dtype=float64)
for i in range(20):
for j in range(10):
array2d[i, j] = float(i + 2*j)/200.0
cosh2d = cosh(5.0 + (array2d))**2.0
verify2d(array2d, cosh2d, 20, 10)
def elemental_cosh_():
i: i32
j: i32
array2d: f64[20, 10] = empty((20, 10), dtype=float64)
cosh2d: f64[20, 10] = empty((20, 10), dtype=float64)
for i in range(20):
for j in range(10):
array2d[i, j] = float(i + 2*j)/200.0
cosh2d = cosh(5.0 + (array2d))**2.0
verify2d(array2d, cosh2d, 20, 10)
def elemental_trig_identity():
i: i32; j: i32; k: i32; l: i32
eps: f64 = 1e-12
arraynd: f64[10, 5, 2, 4] = empty((10, 5, 2, 4), dtype=float64)
identity1: f64[10, 5, 2, 4] = empty((10, 5, 2, 4), dtype=float64)
identity2: f64[10, 5, 2, 4] = empty((10, 5, 2, 4), dtype=float64)
identity3: f64[10, 5, 2, 4] = empty((10, 5, 2, 4), dtype=float64)
identity4: f64[10, 5, 2, 4] = empty((10, 5, 2, 4), dtype=float64)
observed1d_1: f64[400] = empty(400, dtype=float64)
observed1d_2: f64[400] = empty(400, dtype=float64)
observed1d_3: f64[400] = empty(400, dtype=float64)
observed1d_4: f64[400] = empty(400, dtype=float64)
for i in range(10):
for j in range(5):
for k in range(2):
for l in range(4):
arraynd[i, j, k, l] = sin(float(i + j + k + l))
identity1 = 1.0 - cosh(arraynd)**2.0 + sinh(arraynd)**2.0
identity2 = cosh(-1.0 * arraynd) - cosh(arraynd)
identity3 = sinh(-1.0 * arraynd) + sinh(arraynd)
identity4 = (cosh(arraynd/4.0 + arraynd/2.0) -
cosh(arraynd/4.0) * cosh(arraynd/2.0) -
sinh(arraynd/4.0) * sinh(arraynd/2.0))
newshape: i32[1] = empty(1, dtype=int32)
newshape[0] = 400
observed1d_1 = reshape(identity1, newshape)
observed1d_2 = reshape(identity2, newshape)
observed1d_3 = reshape(identity3, newshape)
observed1d_4 = reshape(identity4, newshape)
for i in range(400):
assert abs(observed1d_1[i] - 0.0) <= eps
assert abs(observed1d_2[i] - 0.0) <= eps
assert abs(observed1d_3[i] - 0.0) <= eps
assert abs(observed1d_4[i] - 0.0) <= eps
elemental_sinh()
elemental_cosh()
elemental_trig_identity()