Commit Graph

1343 Commits

Author SHA1 Message Date
lyken c486c469b2
standalone: add test_ndarray_subscript_assignment 2024-08-22 10:08:25 +08:00
lyken b02c34eee9
core/ndstrides: implement ndarray subscript assignment 2024-08-22 10:08:25 +08:00
lyken 16e516b9f7
core/ndstrides: implement cmpop 2024-08-22 10:08:25 +08:00
lyken 913db43e71
core/ndstrides: implement unary op 2024-08-22 10:08:25 +08:00
lyken 6049d34107
core/ndstrides: implement nalgebra functions 2024-08-22 10:08:25 +08:00
lyken 843d25c887
core/ndstrides: add NDArrayObject::to_any 2024-08-22 10:08:25 +08:00
lyken 42cd4e082a
core/ndstrides: add ContiguousNDArray
Currently this is used to interop with nalgebra.
2024-08-22 10:08:25 +08:00
lyken 6cc1055062
core/ndstrides: partially update builtin_fns to use ndarray with strides
nalgebra functions are not updated
2024-08-22 10:08:25 +08:00
lyken bec669ebd3
core/ndstrides: implement np_dot() for scalars and 1D 2024-08-22 10:08:25 +08:00
lyken 90e23caaaf
standalone: extend test_ndarray_matmul 2024-08-22 10:08:25 +08:00
lyken b034bde3e1
core/ndstrides: implement general ndarray matmul 2024-08-22 10:08:25 +08:00
lyken 9ed6fb1cd5
core/ndstrides: implement binop 2024-08-22 10:08:25 +08:00
lyken c1eecbed15
core/ndstrides: add NDArrayOut, broadcast_map and map 2024-08-22 10:08:25 +08:00
lyken 948c933381
core/ndstrides: add more ScalarOrNDArray and NDArrayObject utils 2024-08-22 10:08:25 +08:00
lyken 9a4b2b91e1
core/ndstrides: implement np_transpose() (no axes argument) 2024-08-22 10:08:25 +08:00
lyken a9d87dcfda
core/ndstrides: implement broadcasting & np_broadcast_to() 2024-08-22 10:08:24 +08:00
lyken caf0934d46
core/ndstrides: implement np_reshape() 2024-08-22 10:08:23 +08:00
lyken 6662ff3aca
core: categorize np_{transpose,reshape} as 'view functions' 2024-08-22 10:06:45 +08:00
lyken 00f591629e
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 10:06:45 +08:00
lyken 45ce965220
core/ndstrides: implement ndarray.fill() and .copy() 2024-08-22 10:06:45 +08:00
lyken bed293530d
core/ndstrides: implement np_identity() and np_eye() 2024-08-22 10:06:45 +08:00
lyken 0dc4a37085
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 10:06:45 +08:00
lyken b12394b3dc
core/irrt: add List
Needed for implementing np_array()
2024-08-22 10:06:45 +08:00
lyken 59c2cd6178
core/ndstrides: add NDArrayObject::atleast_nd 2024-08-22 10:06:45 +08:00
lyken cb25e9c035
core/ndstrides: add NDArrayObject::make_copy 2024-08-22 10:06:45 +08:00
lyken 79c7ddf2f4
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
#486.
2024-08-22 10:06:45 +08:00
lyken 7ab8ab5e6a
core/irrt: rename NDIndex to NDIndexInt
The name `NDIndex` is used in later commits.
2024-08-22 10:06:45 +08:00
lyken 199ff1c896
core/irrt: add Slice and Range
Needed for implementing general ndarray indexing
2024-08-22 10:06:45 +08:00
lyken b9b0d83455
core/ndstrides: implement & refactor call_len() 2024-08-22 10:06:45 +08:00
lyken 887d0b1fb6
core/ndstrides: implement ndarray np_{zeros,ones,full,empty} 2024-08-22 10:06:45 +08:00
lyken f30bde2861
core/object: add is_instance for ndarray, tuple and list 2024-08-22 10:06:45 +08:00
lyken 52388eba57
core/model: add util::gen_for_model 2024-08-22 10:06:45 +08:00
lyken 61a593de3e
core/object: add ListObject and TupleObject
Needed for implementing other ndarray utils.
2024-08-22 10:06:45 +08:00
lyken 509c65cf31
core/ndstrides: add ndarray iterator (NDIter) 2024-08-22 10:06:45 +08:00
lyken c4da46baac
core/ndstrides: define ndarray with strides 2024-08-22 10:06:45 +08:00
lyken 3aab374fcf
core/toplevel/helper: add {extract,create}_ndims 2024-08-22 10:06:45 +08:00
lyken 9c50c97d33
core/object: introduce object
Small abstraction to simplify implementations.
2024-08-22 10:06:45 +08:00
lyken df09274d58
core/model: introduce models 2024-08-22 10:06:45 +08:00
lyken 1c7ce22fd5
core/irrt/exceptions: add debug utils with exceptions 2024-08-22 10:06:45 +08:00
lyken 9d3d552e93
core/irrt/exceptions: allow irrt to raise exceptions 2024-08-22 10:06:45 +08:00
lyken d6deb5af26
core/irrt: split irrt.cpp into headers 2024-08-22 10:06:45 +08:00
lyken e3f3c498fe
core/irrt: build.rs capture IR defined constants 2024-08-22 10:06:45 +08:00
lyken 20781d11fd
core/irrt: build.rs capture IR defined types 2024-08-22 10:06:45 +08:00
lyken ecd72e8cad
core/irrt: reformat 2024-08-22 10:06:45 +08:00
lyken b2183fba24
core: add .clang-format 2024-08-22 10:06:45 +08:00
lyken 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
David Mak c5ae0e7c36 [standalone] Add tests for tuple equality 2024-08-21 16:25:32 +08:00
David Mak b8dab6cf7c [standalone] Add tests for string equality 2024-08-21 16:25:32 +08:00
David Mak 4d80ba38b7 [core] codegen/expr: Implement comparison of tuples 2024-08-21 16:25:32 +08:00
David Mak 33929bda24 [core] typecheck/typedef: Add support for tuple methods 2024-08-21 16:25:32 +08:00