Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Implement parallel_for kernel fusion #213

Merged
merged 25 commits into from
Nov 17, 2023
Merged

Implement parallel_for kernel fusion #213

merged 25 commits into from
Nov 17, 2023

Conversation

NaderAlAwar
Copy link
Contributor

No description provided.

@NaderAlAwar
parallel_dispatch: fix formatting and remove comments
9ca13ed
@NaderAlAwar
Fusion: add initial fusion code
ad0fef9
@NaderAlAwar
Parser: add fused style and deepcopy workunit AST to avoid interferin…
a8435d9 
…g with type inference
@NaderAlAwar
Translator: account for fused workunits
b97aa87
@NaderAlAwar
Compiler: fuse multiple workunit objects into one
480b1c2
@NaderAlAwar
ModuleSetup: add support for used workunits
bcee980
@NaderAlAwar
parallel_dispatch: use f-string instead of string addition
a9b9461
@NaderAlAwar
type_inference: add support for fused workunits
ae3c5fd
@NaderAlAwar
ModuleSetup: remove print statement
a50969c
@NaderAlAwar
Runtime: implement compilation and kernel calls for fused workunits
d447977
@NaderAlAwar
Tests: add tests for kernel fusion
182453f
@NaderAlAwar
Compiler: fix issue where parser object was not in scope
e6f8581
@NaderAlAwar
Runtime: fix issue where module setup for fused workunits was not bei…
35a66c8 
…ng hashed properly
@NaderAlAwar
Tests: fix naming errors in fusion tests
d75d323
@NaderAlAwar
mypy: add missing ignore
effc7d0
@NaderAlAwar
PyKokkosMembers: fix access of entity name after type change to str
c09ac00
@NaderAlAwar
args_type_inference: fix issue where subview decorators are not gener…
0a37bea 
…ated due to wrong type check and allow numpy integers in number of iterations
@NaderAlAwar
Views: add space, layout, and trait as fields to Subview class
bf4a54b
@NaderAlAwar
Parser: add full source file AST to PyKokkosEntity to translate other…
d59f24a 
… pykokkos functions with workunits
@NaderAlAwar
Compiler: pass full_AST to fused PyKokkosEntity
580f0c0
@NaderAlAwar
Runtime: fix import order
142a5cb
@NaderAlAwar
Visitors: add support for View arguments to pykokkos functions
a47fd8a
@NaderAlAwar
Translators: add support for pykokkos functions in the standalone wor…
210290c 
…kunit style
@NaderAlAwar
Examples: add workunits version of ExaMiniMD
5af7e08
@NaderAlAwar
Compiler: use first full AST for kernel fusion
6da11f0
@NaderAlAwar NaderAlAwar merged commit ee2f975 into kokkos:main Nov 17, 2023
4 of 5 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@gliga gliga Awaiting requested review from gliga

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
1 participant
@NaderAlAwar