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

1337 Commits

Author SHA1 Message Date
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
lyken 2ae9196540
core/toplevel/helper: add {extract,create}_ndims 2024-08-22 16:23:00 +08:00
lyken 851ae28460
core/object: introduce object
Small abstraction to simplify implementations.
2024-08-22 16:23:00 +08:00
lyken 7c604237b3
core/model: introduce models 2024-08-22 16:23:00 +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
David Mak a8e92212c0 [core] codegen/expr: Implement string equality 2024-08-21 16:25:32 +08:00
David Mak 908271014a [core] typecheck/magic_methods: Add equality methods to string 2024-08-21 16:25:32 +08:00
David Mak c407622f5c [core] codegen/expr: Add compilation error for unsupported cmpop 2024-08-21 15:46:13 +08:00
David Mak d7952d0629 [core] codegen/expr: Fix assertions not generated for -O0 2024-08-21 15:36:54 +08:00
David Mak ca1395aed6 [core] codegen: Remove redundant return 2024-08-21 15:36:54 +08:00
David Mak 7799aa4987 [meta] Do not specify rev in dependency version 2024-08-21 15:36:54 +08:00