Commit Graph

1328 Commits

Author SHA1 Message Date
lyken 4af1a9b350
core/ndstrides: implement nalgebra functions 2024-08-21 11:34:22 +08:00
lyken f91debfceb
core/ndstrides: add NDArrayObject::to_any 2024-08-21 11:34:22 +08:00
lyken 50bd795f50
core/ndstrides: add ContiguousNDArray
Currently this is used to interop with nalgebra.
2024-08-21 11:34:22 +08:00
lyken e0bad42a15
core/ndstrides: partially update builtin_fns to use ndarray with strides
nalgebra functions are not updated
2024-08-21 11:34:22 +08:00
lyken 8a119127e5
core/ndstrides: implement np_dot() for scalars and 1D 2024-08-21 11:34:22 +08:00
lyken 9c9b832519
core/ndstrides: implement general ndarray matmul 2024-08-21 11:34:22 +08:00
lyken 229a2344e3
core/ndstrides: implement binop 2024-08-21 11:34:18 +08:00
lyken cdf6f375e8
core/ndstrides: add NDArrayOut, broadcast_map and map 2024-08-21 11:33:42 +08:00
lyken 4e3938cdcf
core/ndstrides: add more ScalarOrNDArray and NDArrayObject utils 2024-08-21 11:33:20 +08:00
lyken 6ba3dcf62e
core/ndstrides: implement np_transpose() (no axes argument) 2024-08-21 11:33:20 +08:00
lyken 137cc0c5f2
core/ndstrides: implement np_reshape() 2024-08-21 11:33:20 +08:00
lyken 1ed0a67016
core: categorize np_{transpose,reshape} as 'view functions' 2024-08-21 11:33:20 +08:00
lyken 178c81ac70
core/ndstrides: implement np_size() 2024-08-21 11:33:20 +08:00
lyken dfbe334100
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 11:33:20 +08:00
lyken 1e9d2c9749
core/ndstrides: implement ndarray.fill() and .copy() 2024-08-21 11:33:20 +08:00
lyken 36eaedbfb5
core/ndstrides: implement np_identity() and np_eye() 2024-08-21 11:33:20 +08:00
lyken dc513d64b3
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 11:33:20 +08:00
lyken 35cb482f02
core/irrt: add List
Needed for implementing np_array()
2024-08-21 11:33:20 +08:00
lyken 2b5867fb49
core/ndstrides: add NDArrayObject::atleast_nd 2024-08-21 11:33:20 +08:00
lyken 7be28307b7
core/ndstrides: add NDArrayObject::make_copy 2024-08-21 11:33:20 +08:00
lyken 1aa3f468ad
core/ndstrides: implement general ndarray indexing
`...`, `none` are now also supported.
2024-08-21 11:31:50 +08:00
lyken ccc6ed9748
core/irrt: rename NDIndex to NDIndexInt
The name `NDIndex` is used in later commits.
2024-08-21 11:27:03 +08:00
lyken afb08f61cf
core/irrt: add Slice and ResolvedSlice
Needed for implementing general ndarray indexing
2024-08-21 11:27:03 +08:00
lyken 252341b9e8
core/codegen: refactor call_len() & update to ndarray with strides 2024-08-21 11:27:03 +08:00
lyken 3922601ba1
core/ndstrides: implement ndarray np_{zeros,ones,full,empty} 2024-08-21 11:27:00 +08:00
lyken 799821b556
core/object: add is_instance for ndarray, tuple and list 2024-08-21 11:22:43 +08:00
lyken e00a7707bc
core/model: add util::gen_for_model 2024-08-21 11:22:43 +08:00
lyken 5fe884015b
core/object: add ListObject and TupleObject
Needed for implementing other ndarray utils.
2024-08-21 11:22:43 +08:00
lyken 751a319be9
core/ndstrides: add ndarray iterator 2024-08-21 11:22:43 +08:00
lyken f6b80fc40a
core/ndstrides: add basic ndarray utilities 2024-08-21 11:22:43 +08:00
lyken 20db7298aa
core/ndstrides: define ndarray with strides 2024-08-21 11:22:43 +08:00
lyken ca7fdba4fc
core/toplevel/helper: add {extract,create}_ndims 2024-08-21 11:22:43 +08:00
lyken 46064fdfbf
core/object: introduce object
Small abstraction to simplify implementations.
2024-08-21 11:22:43 +08:00
lyken 57144acd25
core/model: introduce models 2024-08-21 11:22:43 +08:00
lyken cc4c2593f7
core/fix: stringify type in expected concrete type error 2024-08-21 11:22:43 +08:00
lyken efefa13b7e
core/irrt/exceptions: add debug utils with exceptions 2024-08-21 11:22:43 +08:00
lyken 59ddf084d2
core/irrt/exceptions: allow irrt to raise exceptions 2024-08-21 11:22:43 +08:00
lyken 84653279bb
core/irrt: split irrt.cpp into headers 2024-08-21 11:22:43 +08:00
lyken 4d79d8e7f4
core/irrt: build.rs capture IR defined constants 2024-08-21 11:22:43 +08:00
lyken 0ef29f06d7
core/irrt: build.rs capture IR defined types 2024-08-21 11:22:43 +08:00
lyken 40568d2e1b
core/irrt: reformat 2024-08-21 11:22:43 +08:00
lyken d2e6119ac5
core: add .clang-format 2024-08-21 11:22:43 +08:00
lyken c9c5c83e6c
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:22:43 +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
Sebastien Bourdeauducq 0fc26df29e flake: update nixpkgs 2024-08-19 23:53:15 +08:00
David Mak 0b074c2cf2 [artiq] symbol_resolver: Set private linkage for constants 2024-08-19 14:41:43 +08:00
Sébastien Bourdeauducq a0f6961e0e cargo: update dependencies 2024-08-19 13:15:03 +08:00
David Mak b1c5c2e1d4 [artiq] Fix RPC of ndarrays to host 2024-08-15 15:41:24 +08:00