Skip to content

Adding the Kokkos backend to main branch - #17

Open
anant37289 wants to merge 91 commits into
mainfrom
gpu_direct_comms
Open

Adding the Kokkos backend to main branch#17
anant37289 wants to merge 91 commits into
mainfrom
gpu_direct_comms

Conversation

@anant37289

@anant37289 anant37289 commented May 17, 2026

Copy link
Copy Markdown
Collaborator

No description provided.

Sh0g0-1758 and others added 30 commits September 23, 2025 16:40
@Sh0g0-1758
nits
1775fb1
@Sh0g0-1758
blech
2f7cd1d
@Sh0g0-1758
Co-authored-by: Anant jain <anant37289@users.noreply.github.com>
1339c21
@Sh0g0-1758
Revert "Co-authored-by: Anant jain <anant37289@users.noreply.github.c…
7a93362 
…om>"

This reverts commit 1339c21.
@Sh0g0-1758
Revert "blech"
6bfabba 
This reverts commit 2f7cd1d.
@Sh0g0-1758
okay
6d298a0
@Sh0g0-1758
ho
62eba90
@Sh0g0-1758
Revert "ho"
e158c20 
This reverts commit 62eba90.
@Sh0g0-1758
hahaha
b1c70ea
@Sh0g0-1758
success
6e62642
@Sh0g0-1758
it works
cfb4dbd
@Sh0g0-1758
use more kokkos
aadeaf7
@Sh0g0-1758
test more functionality on cpu
f74d0ec
@Sh0g0-1758
cmake build switch
1f82f03
@Sh0g0-1758
Revert "cmake build switch"
fd5c8d5 
This reverts commit 1f82f03.
@AdvaitTahilyani
Improved dot
6290e72
@AdvaitTahilyani
Fixed random gen
81c2663
@AdvaitTahilyani
Fixed axpy
d0b1b6d
@Sh0g0-1758
test a working kokkos codegen pipeline
ecd8dc7
@Sh0g0-1758
kokkos backend codegen for primitive ops
1076d96
@Sh0g0-1758
tested functional correctness
e2e9990
@Sh0g0-1758
kokkos backend for custom binary/unary ops
0c3b2a1
@Sh0g0-1758
kokkos backend for custom operation
0264429
@Sh0g0-1758
faster hash, faster strings, better codegen to produce less kernels
81f6b5c
@Sh0g0-1758
isolate codegen into a retargetable class
d5d8035
@Sh0g0-1758
remove further dependence on vec_map
1fae48c
@Sh0g0-1758
isolate execute and generate/compile kernel
2bcca88
@Sh0g0-1758
pre-compile optimization to avoid repeated so file creation for same …
7c16dca 
…kokkos kernel by each charray element
@Sh0g0-1758
fix unary and binary ops by taking fptr from moved object and passing…
beb5627 
… this in the corresponding function calls
@Sh0g0-1758
self contained functionally correct cpu build with kokkos backend
025ba0e
Sh0g0-1758 and others added 30 commits October 14, 2025 16:53
@Sh0g0-1758
pass in broadcasted scalars as a runtime param to reduce number of ke…
fe0e7f2 
…rnels generated
@Sh0g0-1758
nit
d23b616
@anant37289 @Sh0g0-1758
feat: ct backend for AST fusion using decorators (#15)
99bf425 
* changes to support fusing multiple lines

* cleanup

* make matrix logic cleaner

* nits

* nit

* fix carving out regions from instruction list

* creation reordering

---------

Co-authored-by: Sh0g0-1758 <shouryagoel10000@gmail.com>
@Sh0g0-1758 @anant37289
dealloc (#16)
4a26839 
* changes to support fusing multiple lines

* cleanup

* make matrix logic cleaner

* nits

* nit

* fix carving out regions from instruction list

* creation reordering

* gpu fix

* add dealloc op

---------

Co-authored-by: anant <anantjn92004@gmail.com>
@Sh0g0-1758
Revert "dealloc (#16)"
43a9ae3 
This reverts commit 4a26839.
@Sh0g0-1758
Reapply "dealloc (#16)"
91f60ca 
This reverts commit 43a9ae3.
@anant37289
gpu setup script + different deletion stratergy
a3ff151
@anant37289
patched for matrix on GPU
18d0cce
@AdvaitTahilyani
First CI/CD Test
1815280
@AdvaitTahilyani
Improved CI with testing
e6b4da7
@AdvaitTahilyani
Removed unnecessary build step
a1f8ac6
@Sh0g0-1758
use parallel_reduce for sum
8f0e91a
@anant37289
kokkos kernels for cpu
28a6f68
@Sh0g0-1758
nit
462eb65
Anant Jain
multi stream support
67ea654
Anant Jain
async works
fe8d84b
Anant Jain
rest of async
1568d35
@Sh0g0-1758
reduced comms
750f677
Anant Jain
functionally correct integration for comms and multi stream
f6cee7c
Anant Jain
fixes
4dbba44
Anant Jain
integration now works with CPU backend
dedcebd
Anant Jain
something working
25fe3ff
Anant Jain
something working:2
e2d3390
Anant Jain
working gpudirect
e0c03e1
Anant Jain
more working
e09c86c
@Sh0g0-1758
start optimizing gpu direct
6eee5d8
@Sh0g0-1758
presistent buffers for reduce
6c4c646
@Sh0g0-1758
simplifying ast kernel calls
aa9fa68
@Sh0g0-1758
clean and commit
b237c34
@Sh0g0-1758
cpu backend ifdefs
ad1d1d7
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

3 participants

@anant37289 @Sh0g0-1758 @AdvaitTahilyani