M-Labs/nac3
M-Labs/nac3
8
1
Fork 12
Code Issues 98 Pull Requests 19 Releases Wiki Activity
1,437 Commits 32 Branches 0 Tags 21 MiB
27a6f47330
Commit Graph

852 Commits

Author SHA1 Message Date
David Mak
27a6f47330 [core] codegen: Implement construction of unsized ndarrays 
Partially based on f731e604: core/ndstrides: add more ScalarOrNDArray
and NDArrayObject utils.
 2024-12-13 15:23:31 +08:00
David Mak
061747c67b [core] codegen: Implement NDArrayValue::atleast_nd 
Based on 9cfa2622: core/ndstrides: add NDArrayObject::atleast_nd.
 2024-12-13 15:23:31 +08:00
David Mak
dc91d9e35a [core] codegen: Implement ScalarOrNDArray and use it in indexing 
Based on 8f9d2d82: core/ndstrides: implement ndarray indexing.
 2024-12-13 15:23:31 +08:00
David Mak
438943ac6f [core] codegen: Implement indexing for NDArray 
Based on 8f9d2d82: 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-12-13 15:23:31 +08:00
lyken
678e56c95d [core] irrt: rename NDIndex to NDIndexInt 
Unfortunately the name `NDIndex` is used in later commits. Renaming this
typedef to `NDIndexInt` to avoid amending. `NDIndexInt` will be removed
anyway when ndarray strides is completed.
 2024-12-13 15:23:31 +08:00
lyken
fdfc80ca5f [core] codegen: Implement Slice{Type,Value}, RustSlice 
Based on 01c96396: core/irrt: add Slice and Range and part of
8f9d2d82: core/ndstrides: implement ndarray indexing.

Needed for implementing general ndarray indexing.

Currently IRRT slice and range have nothing to do with NAC3's slice
and range. The IRRT slice and range are currently there to implement
ndarray specific features. However, in the future their definitions may
be used to replace that of NAC3's. (NAC3's range is a [i32 x 3], IRRT's
range is a proper struct. NAC3 does not have a slice struct).
 2024-12-13 15:23:31 +08:00
David Mak
8b3429d62a [artiq] Reimplement get_obj_value for strided ndarray 
Based on 7ef93472: artiq: reimplement get_obj_value to use ndarray with
strides
 2024-12-13 15:23:31 +08:00
David Mak
ddd16738a6 [core] codegen: implement ndarray iterator NDIter 
Based on 50f960ab: core/ndstrides: implement ndarray iterator NDIter

A necessary utility to iterate through all elements in a possibly
strided ndarray.
 2024-12-13 15:23:31 +08:00
David Mak
e4bd376587 [core] codegen: Implement ContiguousNDArray 
Fixes compatibility with linalg algorithms. matrix_power is missing due
to the need for indexing support.
 2024-12-13 15:23:29 +08:00
David Mak
44498f22f6 [core] codegen: Implement NDArray functions from a0a1f35b 2024-12-13 15:22:11 +08:00
David Mak
110416d07a [core] codegen/irrt: Add IRRT functions for strided-ndarray 2024-12-13 15:22:11 +08:00
David Mak
08a7d01a13 [core] Add itemsize and strides to NDArray struct 
Temporarily disable linalg ndarray tests as they are not ported to work
with strided-ndarray.
 2024-12-13 15:22:09 +08:00
David Mak
3cd36fddc3 [core] codegen/types: Add check_struct_type_matches_fields 
Shorthand for checking if a type is representable by a StructFields
instance.
 2024-12-12 11:40:44 +08:00
David Mak
56a7a9e03d [core] codegen: Add helper functions for create+call functions 
Replacement for various FnCall methods from legacy ndstrides
implementation.
 2024-12-12 11:30:36 +08:00
David Mak
574ae40f97 [core] codegen: Add call_memcpy_generic_array 
Replacement for Instance<Ptr>::copy_from from legacy ndstrides
implementation.
 2024-12-12 11:30:36 +08:00
David Mak
aa293b6bea [core] codegen: Add type_aligned_alloca 2024-12-12 11:30:35 +08:00
David Mak
eb4b881690 [core] Expose {types,values}::ndarray modules 
Allows better encapsulation of members in these modules rather than
allowing them to leak into types/values mod.
 2024-12-12 11:30:14 +08:00
David Mak
3d0a1d281c [core] Expose irrt::ndarray 2024-12-10 12:49:49 +08:00
David Mak
ad67a99c8f [core] codegen: Cleanup builtin_fns.rs 
- Unpack tuples directly in function argument
- Replace Vec parameters with slices
- Replace unwrap-transform with map-unwrap
 2024-12-10 12:49:49 +08:00
David Mak
8e2b50df21 [core] codegen/ndarray: Cleanup 
- Remove redundant size param
- Add *_fields functions and docs
 2024-12-09 13:01:08 +08:00
David Mak
06092ad29b [core] Move alloca and map_value of ProxyType to implementations 
These functions may not be invokable by the same set of parameters as
some classes has associated states.
 2024-12-09 12:51:50 +08:00
David Mak
d62c6b95fd [core] codegen/types: Rename StructField::set_from_value 2024-12-09 12:51:50 +08:00
David Mak
95e29d9997 [core] codegen: Move ndarray type/value as a separate module 2024-12-09 12:51:46 +08:00
David Mak
536ed2146c [meta] Remove all mentions of build_int_cast 
build_int_cast performs signed extension or truncation depending on the
source and target int lengths. This is usually not what we want - We
want zero-extension instead.

Replace all instances of build_int_cast with
build_int_z_extend_or_bit_cast to fix this issue.
 2024-12-09 12:51:39 +08:00
abdul124
babc081dbd core/toplevel: update tests 2024-11-27 14:31:57 +08:00
abdul124
5337dbe23b core/toplevel: add python-like error messages for class definition 2024-11-27 14:31:57 +08:00
abdul124
f862c01412 core/toplevel: refactor composer 2024-11-27 14:31:53 +08:00
David Mak
28f574282c [core_derive] Ignore doctest in example 
Causes linker errors for unknown reasons.
 2024-11-22 00:00:05 +08:00
David Mak
144f0922db [core] coregen/types: Implement StructFields for NDArray 
Also rename some fields to better align with their naming in numpy.
 2024-11-21 14:27:00 +08:00
David Mak
c58ce9c3a9 [core] codegen/types: Implement NDArray in terms of i8* 
Better aligns with the future implementation of ndstrides.
 2024-11-21 14:27:00 +08:00
David Mak
f7e296da53 [core] irrt: Break IRRT into several impl files 
Each IRRT file is now mapped to one Rust file.
 2024-11-21 14:27:00 +08:00
David Mak
b58c99369e [core] irrt: Update some IRRT implementation 
- Change CSlice to use `void*` for better pointer compatibility
- Only include impl *.hpp files in irrt.cpp
- Refactor typedef to using declaration
- Add missing ``// namespace`
 2024-11-21 14:26:58 +08:00
David Mak
1a535db558 [core] codegen: Add dtype to NDArrayType 
We won't have this once NDArray is refactored to strided impl.
 2024-11-20 15:35:57 +08:00
David Mak
1ba2e287a6 [core] codegen: Add Self::llvm_type to all type abstractions 2024-11-20 15:35:57 +08:00
lyken
f95f979ad3 core/irrt: fix exception.hpp C++ castings 2024-11-20 15:35:57 +08:00
lyken
48e2148c0f core/toplevel/helper: add {extract,create}_ndims 2024-11-20 15:35:57 +08:00
David Mak
88e57f7120 [core_derive] Initial implementation 2024-11-20 15:35:55 +08:00
David Mak
d7633c42bc [core] codegen/types: Implement StructField{,s} 
Loosely based on FieldTraversal by lyken.
 2024-11-19 13:46:25 +08:00
David Mak
a4f53b6e6b [core] codegen: Refactor ProxyType and ProxyValue 
Accepts generator+context object for generic type checking. Also
implements more default trait impl for easier delegation.
 2024-11-19 13:46:25 +08:00
David Mak
9d9ead211e [core] Move Proxies to their own modules 2024-11-19 13:46:23 +08:00
David Mak
26a1b85206 [core] codegen/classes: Remove Underlying type 
This is confusing and we want a better abstraction than this.
 2024-11-19 13:45:55 +08:00
David Mak
2822074b2d [meta] Cleanup from upgrading Rust version 
- Remove rust_2024_edition warnings, since it wouldn't be released for
another 3 months
- Fix new clippy warnings
 2024-11-19 13:43:57 +08:00
Sebastien Bourdeauducq
2cee760404 turn rust_2024_compatibility lints into warnings 2024-11-16 13:41:49 +08:00
occheung
2bd3f63991 boolop: terminate both branches with *_end_bb 2024-11-16 12:06:20 +08:00
occheung
979209a526 binop: expand not operator as loglcal not 2024-11-08 17:12:01 +08:00
David Mak
b6cd2a6993 [meta] Reorganize order of use declarations - Phase 3 2024-10-17 16:25:52 +08:00
David Mak
a98f33e6d1 [meta] Reorganize order of use declarations - Phase 2 
Some more rules:

- For module-level imports, prefer no prefix > super > crate.
- Use crate instead of super if super refers to the crate-level module
 2024-10-17 15:57:33 +08:00
David Mak
56c845aac4 [standalone] Add support for registering globals without type decl 2024-10-07 20:44:06 +08:00
David Mak
65a12d9ab3 [core] Refactor registration of top-level variables 2024-10-07 17:05:48 +08:00
David Mak
9c6685fa8f [core] typecheck/function_check: Fix lookup of defined ids in scope 2024-10-07 16:51:37 +08:00