forked from lcompilers/lpython
-
Notifications
You must be signed in to change notification settings - Fork 0
/
array_02_decl.py
29 lines (22 loc) · 849 Bytes
/
array_02_decl.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
from lpython import i32, i64, f32, f64, c32, c64
from numpy import empty
def accept_multidim_i32_array(xi32: i32[:, :]) -> i32:
return xi32[0, 0]
def accept_multidim_i64_array(xi64: i64[:, :, :]) -> i64:
return xi64[9, 9, 9]
def accept_multidim_f32_array(xf32: f32[:]) -> f32:
return xf32[0]
def accept_multidim_f64_array(xf64: f64[:, :]) -> f64:
return xf64[0, 1]
def declare_arrays():
ai32: i32[3, 3] = empty([3, 3])
ai64: i64[10, 10, 10] = empty([10, 10, 10])
af32: f32[3] = empty(3)
af64: f64[10, 4] = empty([10, 4])
ac32: c32[3, 5, 99] = empty([3, 5, 99])
ac64: c64[10, 13, 11, 16] = empty([10, 13, 11, 16])
print(accept_multidim_i32_array(ai32))
print(accept_multidim_i64_array(ai64))
print(accept_multidim_f32_array(af32))
print(accept_multidim_f64_array(af64))
declare_arrays()