Commit Graph

1332 Commits

Author SHA1 Message Date
lyken b2f2a026b1
core/ndstrides: remove unneeded comment in gen_ndarray_subscript_ndindices 2024-08-21 16:38:57 +08:00
lyken 6b66147dc4
core/ndstrides: remove old ndarray utilities in codegen/numpy.rs 2024-08-21 16:38:57 +08:00
lyken c6cab4f35d
core/ndstrides: implement ndarray subscript assignment 2024-08-21 16:38:57 +08:00
lyken 4194403aaa
core/ndstrides: implement cmpop 2024-08-21 16:38:57 +08:00
lyken a6af84c5f3
core/ndstrides: implement unary op 2024-08-21 16:38:57 +08:00
lyken 1623741c7a
core/ndstrides: implement nalgebra functions 2024-08-21 16:38:57 +08:00
lyken 97817a4750
core/ndstrides: add NDArrayObject::to_any 2024-08-21 16:38:57 +08:00
lyken 8b35d1a4ed
core/ndstrides: add ContiguousNDArray
Currently this is used to interop with nalgebra.
2024-08-21 16:38:57 +08:00
lyken 0e6fe4c596
core/ndstrides: partially update builtin_fns to use ndarray with strides
nalgebra functions are not updated
2024-08-21 16:38:57 +08:00
lyken 5e83f774ea
core/ndstrides: implement np_dot() for scalars and 1D 2024-08-21 16:38:57 +08:00
lyken 685e2cb0cd
core/ndstrides: implement general ndarray matmul 2024-08-21 16:38:57 +08:00
lyken e20a005437
core/ndstrides: implement binop 2024-08-21 16:38:57 +08:00
lyken 2de4084b63
core/ndstrides: add NDArrayOut, broadcast_map and map 2024-08-21 16:38:57 +08:00
lyken 5383ffe332
core/ndstrides: add more ScalarOrNDArray and NDArrayObject utils 2024-08-21 16:38:57 +08:00
lyken 7c335d27a4
core/ndstrides: implement np_transpose() (no axes argument) 2024-08-21 16:38:57 +08:00
lyken fb19fa0729
core/ndstrides: implement np_reshape() 2024-08-21 16:38:56 +08:00
lyken 2b192c8f1a
core: categorize np_{transpose,reshape} as 'view functions' 2024-08-21 16:38:56 +08:00
lyken 3181d430e7
core/ndstrides: implement np_size() 2024-08-21 16:38:56 +08:00
lyken 1d189125e7
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-21 16:38:56 +08:00
lyken 9da0b825d1
core/ndstrides: implement ndarray.fill() and .copy() 2024-08-21 16:38:56 +08:00
lyken 8df7f31fab
core/ndstrides: implement np_identity() and np_eye() 2024-08-21 16:38:56 +08:00
lyken 4c4fe2b52f
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-21 16:38:56 +08:00
lyken 34084e49b7
core/irrt: add List
Needed for implementing np_array()
2024-08-21 16:38:56 +08:00
lyken 4648e7f4a3
core/ndstrides: add NDArrayObject::atleast_nd 2024-08-21 16:38:56 +08:00
lyken 2dfa87c0f7
core/ndstrides: add NDArrayObject::make_copy 2024-08-21 16:38:56 +08:00
lyken 89609909cf
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-21 16:38:56 +08:00
lyken 50356ad01c
core/irrt: rename NDIndex to NDIndexInt
The name `NDIndex` is used in later commits.
2024-08-21 16:38:56 +08:00
lyken 7cdb7a4375
core/irrt: add Slice and Range
Needed for implementing general ndarray indexing
2024-08-21 16:38:55 +08:00
lyken 5990ec6c4f
core/ndstrides: implement & refactor call_len() 2024-08-21 16:29:33 +08:00
lyken 7c09777f96
core/ndstrides: implement ndarray np_{zeros,ones,full,empty} 2024-08-21 16:29:32 +08:00
lyken 91e14b5173
core/object: add is_instance for ndarray, tuple and list 2024-08-21 16:18:11 +08:00
lyken 336bce7fd8
core/model: add util::gen_for_model 2024-08-21 16:18:11 +08:00
lyken 35f15a8305
core/object: add ListObject and TupleObject
Needed for implementing other ndarray utils.
2024-08-21 16:18:11 +08:00
lyken 903e57f0ca
core/ndstrides: add ndarray iterator (NDIter) 2024-08-21 16:18:11 +08:00
lyken 36ff2cc725
core/ndstrides: define ndarray with strides 2024-08-21 16:18:11 +08:00
lyken e4493a7b88
core/toplevel/helper: add {extract,create}_ndims 2024-08-21 16:18:11 +08:00
lyken caccb62563
core/object: introduce object
Small abstraction to simplify implementations.
2024-08-21 16:18:10 +08:00
lyken bef248dad4
core/model: introduce models 2024-08-21 16:18:10 +08:00
lyken 763b6d6c98
core/irrt/exceptions: add debug utils with exceptions 2024-08-21 16:18:08 +08:00
lyken b34673fa9b
core/irrt/exceptions: allow irrt to raise exceptions 2024-08-21 13:17:49 +08:00
lyken f95dc35056
core/irrt: split irrt.cpp into headers 2024-08-21 11:39:20 +08:00
lyken 2fb02884c3
core/irrt: build.rs capture IR defined constants 2024-08-21 11:39:20 +08:00
lyken f7d2e20c7f
core/irrt: build.rs capture IR defined types 2024-08-21 11:39:20 +08:00
lyken b17a8bc6e1
core/irrt: reformat 2024-08-21 11:39:20 +08:00
lyken c19948d198
core: add .clang-format 2024-08-21 11:39:20 +08:00
lyken e979d84ca5
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-21 11:39:20 +08:00
lyken 8532bf5206
standalone: add missing test_ndarray_ceil() run 2024-08-21 11:39:00 +08:00
lyken 2cf64d8608
apply clippy comment changes 2024-08-21 11:21:10 +08:00
lyken 706759adb2
artiq: apply cargo fmt 2024-08-21 11:21:10 +08:00
lyken b90cf2300b
core/fix: add missing lifetime in gen_for* 2024-08-21 11:05:30 +08:00