forked from M-Labs/nac3
1
0
Fork 0
Commit Graph

1358 Commits

Author SHA1 Message Date
lyken 787fe23202
fixup! core/ndstrides: implement ndarray indexing
fix index comment
2024-08-23 11:50:45 +08:00
lyken 5a893e1c15
core: remove Range proxy 2024-08-23 10:57:31 +08:00
lyken 2b29f7b0f5
core: refactor to use RangeObject / Range 2024-08-23 10:47:30 +08:00
lyken 122f55e615
core/object: add RangeObject & RangeObject::len 2024-08-23 10:22:01 +08:00
lyken c0cace843d
core/object: remove *Object::is_instance
This makes implementations awkward.
2024-08-23 10:21:39 +08:00
lyken 5c68ef00ac
core/irrt: refactor out slice::indices and range::len 2024-08-23 10:02:45 +08:00
lyken 3782791323
core: make IRRT slice and range work for any int types 2024-08-23 09:52:08 +08:00
lyken ac6c7c5985
fixup! artiq: reimplement get_obj_value to use ndarray with strides
update symbol_resolver for general array
2024-08-22 20:57:56 +08:00
lyken 8b0305ab6b
fixup! core/model: introduce models
general array
2024-08-22 20:56:25 +08:00
lyken be4b04dbb3
core: remove old ndarray code and NDArray proxy
Nothing depends on the old ndarray implementation now.
2024-08-22 16:34:27 +08:00
lyken e652919b8e
artiq: reimplement get_obj_value to use ndarray with strides 2024-08-22 16:34:27 +08:00
lyken ef391b81aa
artiq: reimplement polymorphic_print for ndarray 2024-08-22 16:34:27 +08:00
lyken ad5506bff1
artiq: reimplement reformat_rpc_arg for ndarrays 2024-08-22 16:34:27 +08:00
lyken 31931b7b26
core/ndstrides: implement np_size() 2024-08-22 16:34:27 +08:00
lyken be6d704020
standalone: add test_ndarray_broadcast_to 2024-08-22 16:23:01 +08:00
lyken 117110dd91
standalone: add test_ndarray_subscript_assignment 2024-08-22 16:23:01 +08:00
lyken c582ffe53d
core/ndstrides: implement ndarray subscript assignment 2024-08-22 16:23:01 +08:00
lyken 51b1921e05
core/ndstrides: implement cmpop 2024-08-22 16:23:01 +08:00
lyken 50457f19e6
core/ndstrides: implement unary op 2024-08-22 16:23:01 +08:00
lyken 54f7e1edfd
core/ndstrides: implement nalgebra functions 2024-08-22 16:23:01 +08:00
lyken 1562a938a1
core/ndstrides: add NDArrayObject::to_any 2024-08-22 16:23:01 +08:00
lyken 00800ba6ee
core/ndstrides: add ContiguousNDArray
Currently this is used to interop with nalgebra.
2024-08-22 16:23:01 +08:00
lyken 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
lyken 566ce7df50
core/ndstrides: implement np_dot() for scalars and 1D 2024-08-22 16:23:01 +08:00
lyken 70c26561e1
standalone: extend test_ndarray_matmul 2024-08-22 16:23:01 +08:00
lyken 4fef633090
core/ndstrides: implement general ndarray matmul 2024-08-22 16:23:01 +08:00
lyken ae351f7678
core/ndstrides: implement binop 2024-08-22 16:23:01 +08:00
lyken 3efae534f7
core/ndstrides: add NDArrayOut, broadcast_map and map 2024-08-22 16:23:01 +08:00
lyken adca310424
core/ndstrides: add more ScalarOrNDArray and NDArrayObject utils 2024-08-22 16:23:01 +08:00
lyken e8e4801c92
core/ndstrides: implement np_transpose() (no axes argument) 2024-08-22 16:23:01 +08:00
lyken 9bdc520bbc
core/ndstrides: implement broadcasting & np_broadcast_to() 2024-08-22 16:23:01 +08:00
lyken 9bd08f8de8
core/ndstrides: implement np_reshape() 2024-08-22 16:23:01 +08:00
lyken 5f94f4a4cd
core: categorize np_{transpose,reshape} as 'view functions' 2024-08-22 16:23:01 +08:00
lyken 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
lyken 99eac99db2
core/ndstrides: implement ndarray.fill() and .copy() 2024-08-22 16:23:01 +08:00
lyken 70e1349d32
core/ndstrides: implement np_identity() and np_eye() 2024-08-22 16:23:01 +08:00
lyken 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
lyken 4dcbaed8ee
core/irrt: add List
Needed for implementing np_array()
2024-08-22 16:23:01 +08:00
lyken 0bb2b02daa
core/ndstrides: add NDArrayObject::atleast_nd 2024-08-22 16:23:01 +08:00
lyken df6916dbe8
core/ndstrides: add NDArrayObject::make_copy 2024-08-22 16:23:01 +08:00
lyken 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
lyken c74b1eb9d8
core/irrt: rename NDIndex to NDIndexInt
The name `NDIndex` is used in later commits.
2024-08-22 16:23:01 +08:00
lyken bda003989e
core/irrt: add Slice and Range
Needed for implementing general ndarray indexing
2024-08-22 16:23:01 +08:00
lyken 5411ac5c88
core/ndstrides: implement & refactor call_len() 2024-08-22 16:23:01 +08:00
lyken 1c317f9205
core/ndstrides: implement ndarray np_{zeros,ones,full,empty} 2024-08-22 16:23:01 +08:00
lyken 701e45364c
core/object: add is_instance for ndarray, tuple and list 2024-08-22 16:23:01 +08:00
lyken fd1a9f4f77
core/model: add util::gen_for_model 2024-08-22 16:23:01 +08:00
lyken f9b8071df9
core/object: add ListObject and TupleObject
Needed for implementing other ndarray utils.
2024-08-22 16:23:00 +08:00
lyken 48fb3ff5ad
core/ndstrides: add ndarray iterator (NDIter) 2024-08-22 16:23:00 +08:00
lyken 4777909543
core/ndstrides: define ndarray with strides 2024-08-22 16:23:00 +08:00