1
0
forked from M-Labs/nac3
Commit Graph

805 Commits

Author SHA1 Message Date
c582ffe53d
core/ndstrides: implement ndarray subscript assignment 2024-08-22 16:23:01 +08:00
51b1921e05
core/ndstrides: implement cmpop 2024-08-22 16:23:01 +08:00
50457f19e6
core/ndstrides: implement unary op 2024-08-22 16:23:01 +08:00
54f7e1edfd
core/ndstrides: implement nalgebra functions 2024-08-22 16:23:01 +08:00
1562a938a1
core/ndstrides: add NDArrayObject::to_any 2024-08-22 16:23:01 +08:00
00800ba6ee
core/ndstrides: add ContiguousNDArray
Currently this is used to interop with nalgebra.
2024-08-22 16:23:01 +08:00
b9e00eb8a5
core/ndstrides: partially update builtin_fns to use ndarray with strides
nalgebra functions are not updated
2024-08-22 16:23:01 +08:00
566ce7df50
core/ndstrides: implement np_dot() for scalars and 1D 2024-08-22 16:23:01 +08:00
4fef633090
core/ndstrides: implement general ndarray matmul 2024-08-22 16:23:01 +08:00
ae351f7678
core/ndstrides: implement binop 2024-08-22 16:23:01 +08:00
3efae534f7
core/ndstrides: add NDArrayOut, broadcast_map and map 2024-08-22 16:23:01 +08:00
adca310424
core/ndstrides: add more ScalarOrNDArray and NDArrayObject utils 2024-08-22 16:23:01 +08:00
e8e4801c92
core/ndstrides: implement np_transpose() (no axes argument) 2024-08-22 16:23:01 +08:00
9bdc520bbc
core/ndstrides: implement broadcasting & np_broadcast_to() 2024-08-22 16:23:01 +08:00
9bd08f8de8
core/ndstrides: implement np_reshape() 2024-08-22 16:23:01 +08:00
5f94f4a4cd
core: categorize np_{transpose,reshape} as 'view functions' 2024-08-22 16:23:01 +08:00
a2a1e74be0
core/ndstrides: implement np_shape() and np_strides()
These functions are not important, but they are handy for debugging +
implementing them takes little effort.

NOTE: `np.strides()` is not an actual NumPy function. You can only(?)
access them thru `ndarray.strides`.
2024-08-22 16:23:01 +08:00
99eac99db2
core/ndstrides: implement ndarray.fill() and .copy() 2024-08-22 16:23:01 +08:00
70e1349d32
core/ndstrides: implement np_identity() and np_eye() 2024-08-22 16:23:01 +08:00
ada6e95d75
core/ndstrides: implement np_array()
It also checks for inconsistent dimensions if the input is a list.
e.g., rejecting `[[1.0, 2.0], [3.0]]`. Previously it was a todo of
`np_array()`.
2024-08-22 16:23:01 +08:00
4dcbaed8ee
core/irrt: add List
Needed for implementing np_array()
2024-08-22 16:23:01 +08:00
0bb2b02daa
core/ndstrides: add NDArrayObject::atleast_nd 2024-08-22 16:23:01 +08:00
df6916dbe8
core/ndstrides: add NDArrayObject::make_copy 2024-08-22 16:23:01 +08:00
0a874da5fc
core/ndstrides: implement ndarray indexing
The functionality for `...` and `np.newaxis` is there in IRRT, but there
is no implementation of them for @kernel Python expressions because of
M-Labs/nac3#486.
2024-08-22 16:23:01 +08:00
c74b1eb9d8
core/irrt: rename NDIndex to NDIndexInt
The name `NDIndex` is used in later commits.
2024-08-22 16:23:01 +08:00
bda003989e
core/irrt: add Slice and Range
Needed for implementing general ndarray indexing
2024-08-22 16:23:01 +08:00
5411ac5c88
core/ndstrides: implement & refactor call_len() 2024-08-22 16:23:01 +08:00
1c317f9205
core/ndstrides: implement ndarray np_{zeros,ones,full,empty} 2024-08-22 16:23:01 +08:00
701e45364c
core/object: add is_instance for ndarray, tuple and list 2024-08-22 16:23:01 +08:00
fd1a9f4f77
core/model: add util::gen_for_model 2024-08-22 16:23:01 +08:00
f9b8071df9
core/object: add ListObject and TupleObject
Needed for implementing other ndarray utils.
2024-08-22 16:23:00 +08:00
48fb3ff5ad
core/ndstrides: add ndarray iterator (NDIter) 2024-08-22 16:23:00 +08:00
4777909543
core/ndstrides: define ndarray with strides 2024-08-22 16:23:00 +08:00
2ae9196540
core/toplevel/helper: add {extract,create}_ndims 2024-08-22 16:23:00 +08:00
851ae28460
core/object: introduce object
Small abstraction to simplify implementations.
2024-08-22 16:23:00 +08:00
7c604237b3
core/model: introduce models 2024-08-22 16:23:00 +08:00
1c7ce22fd5
core/irrt/exceptions: add debug utils with exceptions 2024-08-22 10:06:45 +08:00
9d3d552e93
core/irrt/exceptions: allow irrt to raise exceptions 2024-08-22 10:06:45 +08:00
d6deb5af26
core/irrt: split irrt.cpp into headers 2024-08-22 10:06:45 +08:00
e3f3c498fe
core/irrt: build.rs capture IR defined constants 2024-08-22 10:06:45 +08:00
20781d11fd
core/irrt: build.rs capture IR defined types 2024-08-22 10:06:45 +08:00
ecd72e8cad
core/irrt: reformat 2024-08-22 10:06:45 +08:00
3c4d04e7c9
core/irrt: comment build.rs & move irrt to its own dir
To prepare for future IRRT implementations, and to also make cargo
only have to watch a single directory.
2024-08-22 10:06:45 +08:00
4d80ba38b7 [core] codegen/expr: Implement comparison of tuples 2024-08-21 16:25:32 +08:00
33929bda24 [core] typecheck/typedef: Add support for tuple methods 2024-08-21 16:25:32 +08:00
a8e92212c0 [core] codegen/expr: Implement string equality 2024-08-21 16:25:32 +08:00
908271014a [core] typecheck/magic_methods: Add equality methods to string 2024-08-21 16:25:32 +08:00
c407622f5c [core] codegen/expr: Add compilation error for unsupported cmpop 2024-08-21 15:46:13 +08:00
d7952d0629 [core] codegen/expr: Fix assertions not generated for -O0 2024-08-21 15:36:54 +08:00
ca1395aed6 [core] codegen: Remove redundant return 2024-08-21 15:36:54 +08:00