2242c5af43
core: add linalg methods
2024-07-29 16:48:06 +08:00
72972fa909
core/toplevel: add more numpy categories
2024-07-27 21:57:47 +08:00
142cd48594
core/toplevel: reorder PrimDef::details
2024-07-27 21:57:47 +08:00
8adfe781c5
core/toplevel: fix PrimDef method names
2024-07-27 21:57:47 +08:00
339b74161b
core/toplevel: reorganize PrimDef
2024-07-27 21:57:47 +08:00
05a8948ff2
core: Minor cleanup to use ListValue APIs
2024-07-26 13:35:40 +08:00
6d171ec284
core: Add label name and hooks to gen_for functions
2024-07-26 13:35:40 +08:00
0ba68f6657
core: Set target triple and datalayout for each module
...
Fixes an issue with inconsistent pointer sizes causing crashes.
2024-07-26 13:35:40 +08:00
693b2a8863
core: Add support for 32-bit size_t on 64-bit targets
2024-07-26 13:35:40 +08:00
4948395ca2
core/toplevel/type_annotation: Add handling for mismatching class def
...
Primitive types only contain fields in its Type and not its TopLevelDef.
This causes primitive object types to lack some fields.
2024-07-19 14:42:14 +08:00
3db3061d99
artiq/symbol_resolver: Handle type of zero-length lists
2024-07-19 14:42:14 +08:00
51c2175c80
core/codegen/stmt: Convert assertion values to i1
2024-07-19 14:42:14 +08:00
6c10e3d056
core: cargo clippy
2024-07-12 21:18:53 +08:00
2dbc1ec659
cargo fmt
2024-07-12 21:16:38 +08:00
513d30152b
core: support raise exception short form
2024-07-12 18:58:34 +08:00
2e01b77fc8
core: refactor np_max/np_min functions
2024-07-12 18:18:54 +08:00
cea7cade51
core: add np_argmax/np_argmin functions
2024-07-12 18:18:28 +08:00
eeb474f9e6
core: reduce code duplication in codegen/extern_fns ( #453 )
...
Used macros to reduce code duplication in `codegen/extern_fns`
Reviewed-on: M-Labs/nac3#453
Co-authored-by: abdul124 <ar@m-labs.hk>
Co-committed-by: abdul124 <ar@m-labs.hk>
2024-07-09 16:31:08 +08:00
88b72af2d1
core/llvm_intrinsic: improve macro name and comments
2024-07-09 16:30:32 +08:00
b73f6c4d68
core: reduce code duplication in codegen/llvm_intrinsic
2024-07-09 16:30:32 +08:00
a91602915a
core: Fix missing fields in range type
2024-07-09 13:53:50 +08:00
1c56005a01
core: Reformat and modernize irrt.cpp
...
- Use anon namespace instead of static
- Use using declaration instead of typedef
- Align pointers to the type instead of the identifier
2024-07-09 13:53:50 +08:00
bc40a32524
core: Add report_type_error to enable more code reuse
2024-07-09 13:44:47 +08:00
c820daf5f8
core: Apply cargo format
2024-07-09 13:32:10 +08:00
2cfb7a7e10
core: Refactor range function into constructor
2024-07-09 04:44:40 +08:00
76defac462
meta: use clang -x c++ instead of clang++
2024-07-07 20:03:34 +08:00
650f354b74
core: use C++ for irrt source
2024-07-07 14:36:10 +08:00
f062ef5f59
core/llvm_intrinsic: replace roundeven with rint
2024-07-07 14:24:18 +08:00
f52086b706
core: improve binop and cmpop error messages
2024-07-05 16:27:24 +08:00
0a732691c9
core: refactor typecheck/magic_methods.rs operators & add op symbol name
2024-07-05 16:27:20 +08:00
cbff356d50
core: workaround inkwell on llvm.stackrestore
2024-07-05 13:56:12 +08:00
24ac3820b2
core: check int32 obj_id directly in fold_numpy_function_call_shape_argument
2024-07-05 10:36:47 +08:00
c4052b6342
core: Implement multi-operand __eq__ and __ne__ for lists
2024-07-04 16:01:15 +08:00
66c205275f
core: Implement list::__add__
2024-07-04 16:01:11 +08:00
c85e412206
core: Implement list::__mul__
2024-07-04 15:53:50 +08:00
075536d7bd
core: Add BreakContinueHooks for gen_for_callback
2024-07-04 15:32:18 +08:00
13beeaa2bf
core: Implement handling for zero-length lists
2024-07-04 15:32:18 +08:00
2194dbddd5
core/type_annotation: Refactor List type to TObj
...
In preparation for operators on lists.
2024-07-04 15:32:18 +08:00
d6565feed3
core: ndarray_from_ndlist_impl cast size_of to usize
2024-07-04 12:24:52 +08:00
83154ef8e1
core/llvm_intrinsics: remove llvm.roundeven call from call_float_roundeven
2024-07-03 14:17:47 +08:00
0744b938b8
core: fix __nac3_ndarray_calc_size crash due to incorrect typing
2024-07-03 13:03:14 +08:00
56fa2b6803
core: fix crash on iterating over non-iterables
...
a
2024-06-28 15:45:53 +08:00
d06c13f936
core: fix crash on invalid subscripting
2024-06-27 16:58:48 +08:00
9808923258
core: improve comments in type_inferencer/mod.rs
2024-06-27 14:46:48 +08:00
5b11a1dbdd
core: support tuple and int32 input for np_empty, np_ones, and more
2024-06-27 14:30:17 +08:00
b21df53e0d
core: fix comment typo in unify_call()
2024-06-27 14:06:39 +08:00
0ec967a468
core: improve function call errors
2024-06-27 14:06:39 +08:00
134af79fd6
core: add support for class attributes
2024-06-25 16:06:33 +08:00
7fe2c3496c
core: add attribute field to class definition
2024-06-25 16:06:33 +08:00
144a3fc426
core: more derive Debug in typedef
2024-06-25 15:02:50 +08:00
74096eb9f6
core: name codegen worker threads
2024-06-25 12:36:37 +08:00
06e9d90d57
apply clippy changes
2024-06-21 14:14:01 +08:00
d89146aa02
core: use no_run on builtin_fns docs
2024-06-20 13:53:25 +08:00
0452e6de78
core: Fix codegen for tuple-index into ndarray
2024-06-20 12:50:30 +08:00
635c944c90
core: Fix type inference for tuple-index into ndarray
...
Fixes #420 .
2024-06-20 12:50:30 +08:00
e36af3b0a3
core: reduce code duplication in codegen/builtin_fns ( #422 )
...
Used macros to generate some unary math functions.
Reviewed-on: M-Labs/nac3#422
Reviewed-by: David Mak <chmakac@connect.ust.hk>
Co-authored-by: lyken <lyken@m-labs.hk>
Co-committed-by: lyken <lyken@m-labs.hk>
2024-06-20 12:48:44 +08:00
e85f4f9bd2
core: refactor top_level::builtins::get_builtins()
2024-06-18 11:06:25 +08:00
82091b1be8
meta: Apply clippy changes
2024-06-17 14:10:31 +08:00
2abe75d1f4
core: remove code dup with make_exception_fields
2024-06-17 12:01:48 +08:00
676412fe6d
apply cargo fmt
2024-06-14 09:46:42 +08:00
8b9df7252f
core: cleanup with Unifier::generate_var_id
2024-06-14 09:42:04 +08:00
6979843431
core: fix typo in into_var_map
2024-06-13 16:59:10 +08:00
fed1361c6a
core: rename to_var_map to into_var_map
2024-06-13 16:59:10 +08:00
aa94e0c8a4
core: remove pub & add From<TypeVarId> for u32
2024-06-13 16:59:10 +08:00
f523e26227
core: fix typo in fmt::Display of TypeVarId
2024-06-13 16:59:10 +08:00
f026b48e2a
core: refactor to use TypeVarId and TypeVar
2024-06-13 16:59:10 +08:00
dc874f2994
core: use PrimDef simple names in make_primitives()
2024-06-13 16:58:32 +08:00
23974feae7
meta: Restrict number of allowed lints
2024-06-12 16:10:57 +08:00
40a3bded36
meta: Set clippy lints in {main,lib}.rs
...
So that this does not have to be manually passed to the `cargo clippy`
command-line every single time. Also allows incrementally addressing
these lints by removing and fixing them one-by-one.
2024-06-12 16:10:57 +08:00
c4420e6ab9
core: refactor get_builtins()
2024-06-12 15:09:20 +08:00
fd36f78005
core: refactor PrimitiveDefinitionId into enum PrimDef
2024-06-12 15:01:01 +08:00
8168692cc3
apply cargo fmt
2024-06-12 14:45:03 +08:00
6153f94b05
core/numpy: Implement codegen for np_array
2024-06-11 16:42:11 +08:00
4730b595f3
core/builtins: Add np_array function
2024-06-11 16:42:08 +08:00
c2fdb12397
core/type_inferencer: Add special rule for np_array
2024-06-11 16:40:35 +08:00
82bf14785b
core: Add multidimensional array helpers
2024-06-11 15:30:06 +08:00
2d4329e23c
core/stmt: Use BB of last statement in if-else in phi
2024-06-11 15:30:06 +08:00
679656f9e1
core/classes: Fix incorrect field locations for lists
2024-06-11 15:30:06 +08:00
210d9e2334
core: Add more creator functions for ProxyType
2024-06-11 15:26:37 +08:00
181ac3ec1a
core/classes: Fix incorrect pointers of range.{stop,step}
2024-06-11 15:13:31 +08:00
3acdfb304d
meta: Apply clippy suggestions
2024-06-11 14:58:32 +08:00
f0ab1b858a
core: Refactor class abstractions
...
- Introduce new Type abstractions
- Rearrange some functions
2024-06-06 13:45:51 +08:00
08129cc635
nac3core: add TopLevelComposer::new builtin check's assertion msg
2024-06-05 15:30:02 +08:00
ad4832dcf4
core: Refactor to get LLVM intrinsics via Intrinsics::find
2024-06-05 15:29:40 +08:00
b857f1e403
nac3core: fix typo in gen_for's comment
2024-06-04 17:15:41 +08:00
ed79d5bb9e
core/expr: Add support for multi-dim slicing of NDArrays
2024-06-03 16:40:05 +08:00
c35ad06949
core/expr: Add support for 1D slicing of NDArrays
2024-06-03 16:40:05 +08:00
135ef557f9
core/numpy: Implement ndarray_sliced_{copy,copyto_impl}
...
Performing copying with optional support for slicing. Also made
copy_impl delegate to sliced_copy, as sliced_copy now performs a
superset of operations that copy_impl can already do.
2024-06-03 16:40:05 +08:00
a176c3eb70
core/irrt: Change handle_slice_indices to instead take length of object
...
So that all other array-like datatypes (e.g. ndarray) can also take
advantage of it.
2024-06-03 16:40:05 +08:00
2cf79510c2
core/numpy: Add more helper functions
2024-06-03 16:40:05 +08:00
b6ff75dcaf
core/irrt: Add support for calculating partial size of NDArray
2024-06-03 16:40:05 +08:00
588c15f80d
core/stmt: Add gen_for_range_callback
...
For generating for loops over range objects or array slices.
2024-06-03 16:40:05 +08:00
82cc693b11
meta: Update dependencies
2024-06-03 16:40:02 +08:00
520e1adc56
core/builtins: Add np_minimum/np_maximum
2024-05-09 15:01:20 +08:00
73e81259f3
core/builtins: Add np_min/np_max
2024-05-09 15:01:20 +08:00
7627acea41
core/type_inferencer: Fix error message
2024-05-09 15:01:20 +08:00
a777099ea8
core/type_inferencer: Fix missing lowering for some builtin TVars
2024-05-09 15:01:20 +08:00
30c6cffbad
core/builtins: Refactored numpy builtins to accept scalar and ndarrays
2024-05-06 15:38:29 +08:00
51671800b6
core/builtins: Extract codegen portion into functions
...
We will need to reuse them when implementing elementwise function
application for ndarrays.
2024-05-06 13:21:54 +08:00
7195476edb
core/builtins: Add llvm_intrinsics prefix
2024-05-06 13:21:54 +08:00
eecba0b71d
core: Add GenCall::create_dummy
...
A simple abstraction for GenCalls that are already handled elsewhere.
2024-05-06 13:21:54 +08:00
7b4253ccd8
core/numpy: Add missing lifetime parameters
2024-05-06 13:21:54 +08:00
f58c3a11f8
core/builtins: Rework handling of PrimitiveStore-Unifier tuples
2024-05-06 13:21:54 +08:00
d0766a116f
core: Remove Box from GenCallCallback type alias
...
So that references to the function type can be taken.
2024-05-06 13:21:54 +08:00
64a3751fc2
core: Remove custom function type definitions for ndarray operators
2024-05-06 13:21:54 +08:00
062e318dd5
core/magic_methods: Fix clippy warnings
2024-05-06 13:21:54 +08:00
847615fc2f
core: Implement numpy.matmul for 2D-2D ndarrays
2024-04-23 10:27:37 +08:00
5dfcc63978
core/classes: Take reference of indexes
2024-04-16 17:20:24 +08:00
025b3cd02f
core/stmt: Remove gen_if_chained*
...
Turns out it is really difficult to get lifetimes and closures right, so
let's just provide the most rudimentary if-else codegen and we can nest
them if necessary.
2024-04-16 17:16:50 +08:00
e0f440040c
core/expr: Implement negative indices for ndarray
2024-04-15 12:49:42 +08:00
f0715e2b6d
core/stmt: Add gen_if* functions
...
For generating if-constructs in IR.
2024-04-15 12:20:34 +08:00
e7fca67786
core/stmt: Do not generate jumps if bb is already terminated
...
Future-proofs gen_*_callback functions in case other codegen functions
will delegate to it in the future.
2024-04-15 12:20:34 +08:00
52c731c312
core: Implement Not/UAdd/USub for booleans
...
Not sure if this is deliberate or an oversight, but we implement it
anyway for consistency with other Python implementations.
2024-04-12 18:29:58 +08:00
00d1b9be9b
core: Fix __inv__ for i8-based boolean operands
2024-04-12 15:35:54 +08:00
e614dd4257
core/type_inferencer: Fix location of unary/compare expressions
...
Codegen uses this location information to determine the CallId, and if
a function call is the operand of a unary expression or left-hand
operand of a compare expression, codegen will use the type of the
operator expression rather than the actual operand type.
2024-04-05 15:42:10 +08:00
937a8b9698
core/magic_methods: Fix type of unary ops with primitive types
2024-04-05 13:23:08 +08:00
876ad6c59c
core/type_inferencer: Include location info if inferencer fails
2024-04-05 13:22:35 +08:00
a920fe0501
core: Implement elementwise comparison operators
2024-04-03 00:07:33 +08:00
727a1886b3
core: Implement elementwise unary operators
2024-04-03 00:07:33 +08:00
6af13a8261
core: Implement elementwise binary operators
...
Including immediate variants of these operators.
2024-04-03 00:07:33 +08:00
3540d0ab29
core/magic_methods: Add typeof_*op
...
Used to determine the expected type of the binary operator with
primitive operands.
2024-04-03 00:07:33 +08:00
3a6c53d760
core/toplevel/numpy: Split ndarray type var utilities
2024-04-03 00:07:33 +08:00
87bc34f7ec
core: Implement calculations for broadcasting ndarrays
2024-04-03 00:07:31 +08:00
f50a5f0345
core/type_inferencer: Allow both int32 and isize when indexing ndarray
2024-04-02 16:49:12 +08:00
a77fd213e0
core/magic_methods: Allow unknown return types
...
These types can be later inferred by the type inferencer.
2024-04-02 16:49:12 +08:00
8f1497df83
core/helper: Add PrimitiveDefinitionIds::iter
2024-04-02 16:49:12 +08:00
5ca2dbeec8
core/typedef: Add Type::obj_id to replace get_obj_id
2024-04-02 16:49:10 +08:00
9a98cde595
core: Extract codegen portion of gen_*op_expr
...
This allows *ops to be generated internally using LLVM values as
input. Required in a future change.
2024-04-01 16:48:25 +08:00
5ba8601b39
core: Remove ArrayValue variants of functions
...
These will be lowered and optimized away later anyways, and we have
ArrayLikeAccessor now.
2024-04-01 16:48:25 +08:00
26a01b14d5
core: Use more typed slices in APIs
2024-04-01 16:48:25 +08:00
d5f4817134
core/builtins: Fix len() on ndarrays
2024-04-01 16:48:24 +08:00
789bfb5a26
core: Fix index-based operations not returning i32
2024-04-01 16:46:45 +08:00
4bb0e60981
core: Apply clippy suggestions
2024-04-01 16:46:41 +08:00
13f06f3e29
core: Refactor VarMap to IndexMap
...
This is the only Map I can find that preserves insertion order while
also deduplicating elements by key.
2024-03-22 15:51:23 +08:00
f0da9c0283
core: Add ArrayLikeValue
...
For exposing LLVM values that can be accessed like an array.
2024-03-22 15:51:06 +08:00
2c4bf3ce59
core: Allow unsized CodeGenerator to be passed to some codegen functions
...
Enables codegen_callback to call these codegen functions as well.
2024-03-22 15:07:28 +08:00
e980f19c93
core: Simplify typed value assertions
2024-03-22 15:07:28 +08:00
cfbc37c1ed
core: Add gen_for_callback_incrementing
...
Simplifies generation of monotonically increasing for loops.
2024-03-22 15:07:28 +08:00
50264e8750
core: Add missing unchecked accessors for NDArrayDimsProxy
2024-03-22 15:07:28 +08:00
1b77e62901
core: Split numpy into codegen and toplevel
2024-03-22 15:07:28 +08:00
fd44ee6887
core: Apply clippy suggestions
2024-03-22 15:07:23 +08:00
c8866b1534
core/classes: Rename get_* functions to remove prefix
...
As suggested by Rust API Guidelines.
2024-03-21 15:46:10 +08:00
84a888758a
core: Rename unsafe functions to unchecked
...
This is this intended name of the functions.
2024-03-21 15:46:10 +08:00
c3b122acfc
core: Implement ndarray.copy
2024-03-07 14:59:13 +08:00
a94927a11d
core: Update __builtin_assume expressions
...
No dimension size should be 0.
2024-03-07 14:59:13 +08:00
ebf86cd134
core: Use size_t for accessing array elements
2024-03-07 14:59:13 +08:00
cccd8f2d00
core: Fix ndarray_eye not preserving signness of offset
2024-03-07 14:59:13 +08:00
3292aed099
core: Fix ndarray subscript operator returning the wrong object
...
Should be returning the newly created object instead of the original
ndarray...
2024-03-07 14:59:13 +08:00
96b7f29679
core: Implement ndarray.fill
2024-03-07 14:59:13 +08:00
3d2abf73c8
core: Replace ndarray_init_dims IRRT impl with IR impl
...
Implementation of that function in IR allows for more flexibility in
terms of different integer type widths.
2024-03-07 14:59:13 +08:00
