Commit Graph

1357 Commits

Author SHA1 Message Date
lyken 58be4a8b09
core: add RustSlice::indices 2024-08-25 20:58:56 +08:00
lyken 52da6347ee
fixup! core: move Slice and RustSlice to its own file & genericize their IntKind 2024-08-25 20:40:28 +08:00
lyken a1410833bc
core: move gen_slice to object/slice.rs and refactor 2024-08-25 20:28:34 +08:00
lyken b38d9000f8
core: add RustRange 2024-08-25 20:28:30 +08:00
lyken a6e1d354b6
core: move Slice and RustSlice to its own file & genericize their IntKind 2024-08-25 20:19:07 +08:00
lyken f698b0c1fa
core/model: refactor CSlice and Exception with models 2024-08-25 17:01:44 +08:00
lyken 8eb9094d68
fix Struct const_struct comment 2024-08-25 17:01:44 +08:00
lyken 3f322de9a0
fix: model CheckTypeFieldTraversal comment 2024-08-25 17:01:44 +08:00
lyken 6528115d6a
core: refactor range with model and RangeObject 2024-08-25 17:01:44 +08:00
lyken b13b28cfe8
core: update insta after ndstrides
New type vars are introduced when programming new ndarray functions.
2024-08-25 16:58:27 +08:00
lyken 6c48adff4d
core/model/ptr: renaming and add notes on upgrading to LLVM 15 2024-08-25 14:21:46 +08:00
lyken 7701691794
core: remove old ndarray code and NDArray proxy
Nothing depends on the old ndarray implementation now.
2024-08-25 00:53:04 +08:00
lyken bca3313ad6
artiq: reimplement get_obj_value to use ndarray with strides 2024-08-25 00:52:41 +08:00
lyken 6d2ac3bd8a
artiq: reimplement polymorphic_print for ndarray 2024-08-25 00:52:41 +08:00
lyken b91994fcab
artiq: reimplement reformat_rpc_arg for ndarray 2024-08-25 00:52:41 +08:00
lyken 881cbc5142
standalone/ndarray: improve {reshape,broadcast_to,transpose} tests
Print their shapes and exhaustively print all contents.
2024-08-25 00:52:41 +08:00
lyken 55965dda9e
standalone/ndarray: add and organize view function tests 2024-08-25 00:52:41 +08:00
lyken 6285e67f3e
core/ndstrides: update builtin_fns to use ndarray with strides 2024-08-25 00:52:41 +08:00
lyken 9200ba2521
core/ndstrides: add NDArrayObject::to_any 2024-08-25 00:52:41 +08:00
lyken 040948543b
core/ndstrides: add ContiguousNDArray
Currently this is used to interop with nalgebra.
2024-08-25 00:52:41 +08:00
lyken 1165ac5e90
core/ndstrides: implement np_dot() for scalars and 1D 2024-08-25 00:52:41 +08:00
lyken 61ea014c61
core/ndstrides: implement general matmul 2024-08-25 00:52:41 +08:00
lyken fe3405f50d
core/ndstrides: implement cmpop 2024-08-25 00:52:41 +08:00
lyken 09875a2555
core/ndstrides: implement unary op 2024-08-25 00:52:41 +08:00
lyken 56e448a040
core/ndstrides: implement binop 2024-08-25 00:52:41 +08:00
lyken 58178223e7
core/ndstrides: add NDArrayOut, broadcast_map and map 2024-08-25 00:52:41 +08:00
lyken 1a4a6bc5f9
core/ndstrides: implement subscript assignment 2024-08-25 00:52:41 +08:00
lyken 19ee579226
core/ndstrides: add more ScalarOrNDArray and NDArrayObject utils 2024-08-25 00:52:41 +08:00
lyken 75ba01d547
core/ndstrides: implement np_transpose() (no axes argument)
The IRRT implementation knows how to handle axes. But the argument is
not in NAC3 yet.
2024-08-25 00:52:41 +08:00
lyken 25be81cc83
core/ndstrides: implement broadcasting & np_broadcast_to() 2024-08-25 00:52:41 +08:00
lyken 34709cf076
core/ndstrides: implement np_reshape() 2024-08-25 00:52:41 +08:00
lyken d37d0c8da4
core: categorize np_{transpose,reshape} as 'view functions' 2024-08-25 00:52:41 +08:00
lyken ee66cac8c2
core/ndstrides: implement np_size() 2024-08-25 00:52:41 +08:00
lyken 854a3eb6f0
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-25 00:52:41 +08:00
lyken 72c2b9d840
core/ndstrides: implement ndarray.fill() and .copy() 2024-08-25 00:52:41 +08:00
lyken a004703ec2
core/ndstrides: implement np_identity() and np_eye() 2024-08-25 00:52:41 +08:00
lyken 0ec4d13735
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-25 00:52:41 +08:00
lyken dd1a19d97f
core/irrt: add List
Needed for implementing np_array()
2024-08-25 00:52:41 +08:00
lyken ea1410f9ff
core/ndstrides: add NDArrayObject::atleast_nd 2024-08-25 00:52:41 +08:00
lyken b175409a9b
core/ndstrides: add NDArrayObject::make_copy 2024-08-25 00:52:41 +08:00
lyken 7e22f09e6e
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-25 00:52:41 +08:00
lyken 494616f0d9
core/irrt: rename NDIndex to NDIndexInt
The name `NDIndex` is used in later commits.
2024-08-25 00:52:41 +08:00
lyken 2bbc619151
core/irrt: add Slice and Range
Needed for implementing general ndarray indexing.

Currently the IRRT slice and range have nothing to do with NAC3's slice
and range.
2024-08-25 00:52:41 +08:00
lyken e0d1e9a007
core/ndstrides: implement len(ndarray) & refactor len() 2024-08-25 00:52:41 +08:00
lyken 4929dcbf79
core/ndstrides: implement np_{zeros,ones,full,empty} 2024-08-25 00:52:41 +08:00
lyken 3d4c6b6ac0
core/model: add util::gen_for_model 2024-08-25 00:52:41 +08:00
lyken c5aa1616f8
core/object: add ListObject and TupleObject
Needed for implementing other ndarray utils.
2024-08-25 00:52:41 +08:00
lyken 98b65cf3e2
core/ndstrides: implement ndarray iterator NDIter 2024-08-25 00:52:41 +08:00
lyken 4a26a0e222
core/ndstrides: introduce NDArray
NDArray with strides.
2024-08-25 00:52:41 +08:00
lyken e927d1a308
core/toplevel/helper: add {extract,create}_ndims 2024-08-25 00:52:41 +08:00