Commit Graph

803 Commits

Author SHA1 Message Date
lyken 911d0accc2
core/ndstrides: add ContiguousNDArray
Currently this is used to interop with nalgebra.
2024-08-26 11:18:24 +08:00
lyken 75b2e80418
core/ndstrides: implement np_dot() for scalars and 1D 2024-08-26 11:18:24 +08:00
lyken b416ece921
core/ndstrides: implement general matmul 2024-08-26 11:18:24 +08:00
lyken 9bf0e2cbf4
core/ndstrides: implement cmpop 2024-08-26 11:18:23 +08:00
lyken 5f143d2f2f
core/ndstrides: implement unary op 2024-08-26 11:18:23 +08:00
lyken f5698a9eed
core/ndstrides: implement binop 2024-08-26 11:18:23 +08:00
lyken cb8cea4286
core/ndstrides: add NDArrayOut, broadcast_map and map 2024-08-26 11:18:23 +08:00
lyken ccc8ce5886
core/ndstrides: implement subscript assignment 2024-08-26 11:18:23 +08:00
lyken 48ce2d6c8a
core/ndstrides: add more ScalarOrNDArray and NDArrayObject utils 2024-08-26 11:18:23 +08:00
lyken 7a7a67b522
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-26 11:18:23 +08:00
lyken f37b092947
core/ndstrides: implement broadcasting & np_broadcast_to() 2024-08-26 11:18:23 +08:00
lyken 813dad4ed0
core/ndstrides: implement np_reshape() 2024-08-26 11:18:23 +08:00
lyken 48d7032b5e
core: categorize np_{transpose,reshape} as 'view functions' 2024-08-26 11:18:23 +08:00
lyken 6fb988a1e4
core/ndstrides: implement np_size() 2024-08-26 11:18:23 +08:00
lyken 3ea5ffe5ca
core/ndstrides: implement np_shape() and np_strides()
These functions are not important, but they are handy for debugging.

`np.strides()` is not an actual NumPy function, but `ndarray.strides` is used.
2024-08-26 11:18:23 +08:00
lyken fb11b91d09
core/ndstrides: implement ndarray.fill() and .copy() 2024-08-26 11:18:23 +08:00
lyken 9742f795d5
core/ndstrides: implement np_identity() and np_eye() 2024-08-26 11:18:23 +08:00
lyken b158ec80b4
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]]`.
2024-08-26 11:18:23 +08:00
lyken da23bb1417
core/irrt: add List
Needed for implementing np_array()
2024-08-26 11:18:23 +08:00
lyken 9c5273ae09
core/ndstrides: add NDArrayObject::atleast_nd 2024-08-26 11:18:23 +08:00
lyken 3d734aef17
core/ndstrides: add NDArrayObject::make_copy 2024-08-26 11:18:23 +08:00
lyken 510dbfc70e
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-26 11:18:23 +08:00
lyken fda7f8d827
core/irrt: rename NDIndex to NDIndexInt
The name `NDIndex` is used in later commits.
2024-08-26 11:18:23 +08:00
lyken 5537645395
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-26 11:18:23 +08:00
lyken ad5afb52c4
core/ndstrides: implement len(ndarray) & refactor len() 2024-08-26 11:18:23 +08:00
lyken 3a241acc9c
core/ndstrides: implement np_{zeros,ones,full,empty} 2024-08-26 11:18:23 +08:00
lyken b41cc79c05
core/model: add util::gen_for_model 2024-08-26 11:18:23 +08:00
lyken 2c276fa75e
core/object: add ListObject and TupleObject
Needed for implementing other ndarray utils.
2024-08-26 11:18:23 +08:00
lyken f5827dae24
core/ndstrides: implement ndarray iterator NDIter 2024-08-26 11:18:23 +08:00
lyken a531d0127a
core/ndstrides: introduce NDArray
NDArray with strides.
2024-08-26 11:18:23 +08:00
lyken 3d6565d0bf
core/toplevel/helper: add {extract,create}_ndims 2024-08-26 11:18:23 +08:00
lyken 56f44086d6
core/object: introduce object
A small abstraction to simplify implementations.
2024-08-26 11:18:23 +08:00
lyken 714165e00d
core/model: renaming and add notes on upgrading Ptr to LLVM 15 2024-08-26 11:18:20 +08:00
lyken 58a3b100b4
core/model: introduce models 2024-08-26 11:18:15 +08:00
lyken f169d37074
core/irrt/exceptions: add debug utils with exceptions 2024-08-26 09:39:41 +08:00
lyken cf34002179
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`.
2024-08-26 09:38:56 +08:00
lyken 115fff9e65
core/irrt: build.rs capture IR defined constants 2024-08-26 09:35:17 +08:00
lyken 342206989e
core/irrt: build.rs capture IR defined types 2024-08-26 09:35:17 +08:00
lyken e9629a6688
core/irrt: split irrt.cpp into headers 2024-08-26 09:35:06 +08:00
lyken 331ab8a946
core/irrt: reformat 2024-08-26 09:35:00 +08:00
lyken 8f726ecbfa
core/irrt: comment build.rs & move irrt to nac3core/irrt 2024-08-26 09:34:27 +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
David Mak 76016a26ad [meta] Apply clippy suggestions 2024-08-21 13:07:57 +08:00