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

1342 Commits

Author SHA1 Message Date
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
M-Labs/nac3#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
lyken b6aae87bd3
core/object: introduce object
Small abstraction to simplify implementations.
2024-08-25 00:52:41 +08:00
lyken ff84c9bfaf
core/model: introduce models 2024-08-25 00:52:41 +08:00
lyken d75c5d460f
core/irrt/exceptions: add debug utils with exceptions 2024-08-24 15:35:00 +08:00
lyken cd69fa07a5
core/irrt/exceptions: allow irrt to raise exceptions
Achieved through defining all the needed Exception ID constants at link
time.

Secondly, since `Exception` is `size_t` dependent, `__nac3_raise()`
takes an opaque pointer to `Exception`, unless IRRT is compiled into
32-bit and 64-bit separately with a defined `size_t` type.
2024-08-24 15:29:04 +08:00
lyken 82671a4be7
core/irrt: build.rs capture IR defined constants 2024-08-24 15:29:04 +08:00
lyken 176cb992eb
core/irrt: build.rs capture IR defined types 2024-08-24 15:29:04 +08:00
lyken b743603a97
core/irrt: split irrt.cpp into headers 2024-08-24 15:29:04 +08:00
lyken 27a1fc7024
core/irrt: reformat 2024-08-24 15:29:04 +08:00
lyken b69d527752
core: add .clang-format 2024-08-24 15:29:04 +08:00
lyken 2c62b61363
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-24 13:04:06 +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