Repository navigation

feat: add kokkos backend through codegen - #12

Open

Sh0g0-1758 wants to merge 81 commits into

mainfrom

kokkos_back

Open

feat: add kokkos backend through codegen#12
Sh0g0-1758 wants to merge 81 commits into
mainfrom
kokkos_back

Commits on Sep 23, 2025
- nits
  Sh0g0-1758
  committed
  1775fb1
- blech
  Sh0g0-1758
  committed
  2f7cd1d
- Co-authored-by: Anant jain <anant37289@users.noreply.github.com>
  Sh0g0-1758
  committed
  1339c21
- Revert "Co-authored-by: Anant jain <anant37289@users.noreply.github.com>"
  Sh0g0-1758
  committed
  7a93362
- Revert "blech"
  Sh0g0-1758
  committed
  6bfabba
- okay
  Sh0g0-1758
  committed
  6d298a0
- ho
  Sh0g0-1758
  committed
  62eba90
- Revert "ho"
  Sh0g0-1758
  committed
  e158c20
- hahaha
  Sh0g0-1758
  committed
  b1c70ea
- success
  Sh0g0-1758
  committed
  6e62642
- it works
  Sh0g0-1758
  committed
  cfb4dbd
Commits on Sep 24, 2025
- use more kokkos
  Sh0g0-1758
  committed
  aadeaf7
- test more functionality on cpu
  Sh0g0-1758
  committed
  f74d0ec
Commits on Sep 27, 2025
- cmake build switch
  Sh0g0-1758
  committed
  1f82f03
- Revert "cmake build switch"
  Sh0g0-1758
  committed
  fd5c8d5
Commits on Sep 29, 2025
- Improved dot
  AdvaitTahilyani
  committed
  6290e72
- Fixed random gen
  AdvaitTahilyani
  committed
  81c2663
- Fixed axpy
  AdvaitTahilyani
  committed
  d0b1b6d
- test a working kokkos codegen pipeline
  Sh0g0-1758
  committed
  ecd8dc7
- kokkos backend codegen for primitive ops
  Sh0g0-1758
  committed
  1076d96
- tested functional correctness
  Sh0g0-1758
  committed
  e2e9990
Commits on Sep 30, 2025
- kokkos backend for custom binary/unary ops
  Sh0g0-1758
  committed
  0c3b2a1
- kokkos backend for custom operation
  Sh0g0-1758
  committed
  0264429
- faster hash, faster strings, better codegen to produce less kernels
  Sh0g0-1758
  committed
  81f6b5c
- isolate codegen into a retargetable class
  Sh0g0-1758
  committed
  d5d8035
- remove further dependence on vec_map
  Sh0g0-1758
  committed
  1fae48c
- isolate execute and generate/compile kernel
  Sh0g0-1758
  committed
  2bcca88
Commits on Oct 1, 2025
Commits on Oct 2, 2025
- hardware agnostic build
  Sh0g0-1758
  committed
  0b83ed7
- add gpu link options
  Sh0g0-1758
  committed
  fdd0c1e
- fix cmake dependence issue
  Sh0g0-1758
  committed
  53bc7a0
- fix gpu linking
  Sh0g0-1758
  committed
  93de0c2
- fix multi pe error: load library on each pe through group
  Sh0g0-1758
  committed
  2c81aa1
- Missing Includes on MacOS
  AdvaitTahilyani
  committed
  0d59042
- Switched to kokkos 2d
  AdvaitTahilyani
  committed
  e4dc7da
- Updated to use kokkos instead of eigen
  AdvaitTahilyani
  committed
  45e2aa2
Commits on Oct 5, 2025
- generalize codegen to nD views
  Sh0g0-1758
  committed
  9b536cc
- codegen for matrix
  Sh0g0-1758
  committed
  c969261
- fix indexing issue, fix pup issue and run correctness checks
  Sh0g0-1758
  committed
  a30231f
- add indexing to custom unops/binops
  Sh0g0-1758
  committed
  213b8b0
- cleanup
  Sh0g0-1758
  committed
  c1f1391
- Fixed issues for GPU compatibility
  AdvaitTahilyani
  committed
  becd705
Commits on Oct 6, 2025
- use concepts to specialize operators and disambiguate them from arith ops of types that are not vector / matrix
  Sh0g0-1758
  committed
  0c5c51c
- fix indexing scheme for where
  Sh0g0-1758
  committed
  fe4d81b
Commits on Oct 8, 2025
- C++14 type_traits
  Sh0g0-1758
  committed
  c8c3dea
Commits on Oct 9, 2025
- fix vec type check for binop
  Sh0g0-1758
  committed
  873607f
Commits on Oct 12, 2025
- aadding direct ops
  Anant Jain
  committed
  62e0f6c
- clang fmt + lib from lib64 + restore vector.cpp + get name instead of hash
  anant37289
  committed
  5724da3
- ctors for python AST -> flattened AST conversion
  Sh0g0-1758
  committed
  0378b38
- Merge pull request #14 from anant37289/ops_direct
  Sh0g0-1758
  authored
  882915f
Commits on Oct 13, 2025
- Fixed dot
  AdvaitTahilyani
  committed
  7804b2f
- GPU Fix
  AdvaitTahilyani
  committed
  ae98fc3
- fix unary/binary for charmnumerics
  Sh0g0-1758
  committed
  f537beb
- fix where op indexing
  Sh0g0-1758
  committed
  1fc5913
- remove bugs in unops, add print for binops
  anant37289
  committed
  91387a2
- expose matmul as a separate op
  Sh0g0-1758
  committed
  c45e9b2
- fix matmul constructor by actually allocating space for it in the map
  Sh0g0-1758
  committed
  5f923e6
- log base support
  anant37289
  committed
  915dcda
Commits on Oct 14, 2025
- make axpy a custom binary operation
  anant37289
  committed
  7ce8496
- pass in broadcasted scalars as a runtime param to reduce number of kernels generated
  Sh0g0-1758
  committed
  fe0e7f2
- nit
  Sh0g0-1758
  committed
  d23b616
Commits on Oct 15, 2025
- feat: ct backend for AST fusion using decorators (#15)
  
  anant37289
  and
  Sh0g0-1758
  authored
  99bf425
Commits on Oct 17, 2025
- dealloc (#16)
  
  Sh0g0-1758
  and
  anant37289
  authored
  4a26839
- Revert "dealloc (#16)"
  Sh0g0-1758
  committed
  43a9ae3
Commits on Oct 18, 2025
- Reapply "dealloc (#16)"
  Sh0g0-1758
  committed
  91f60ca
Commits on Oct 19, 2025
- gpu setup script + different deletion stratergy
  anant37289
  committed
  a3ff151
Commits on Oct 22, 2025
- patched for matrix on GPU
  anant37289
  committed
  18d0cce
Commits on Oct 23, 2025
- First CI/CD Test
  AdvaitTahilyani
  committed
  1815280
- Improved CI with testing
  AdvaitTahilyani
  committed
  e6b4da7
- Removed unnecessary build step
  AdvaitTahilyani
  committed
  a1f8ac6
Commits on Oct 28, 2025
- use parallel_reduce for sum
  Sh0g0-1758
  committed
  8f0e91a
Commits on Oct 29, 2025
- kokkos kernels for cpu
  anant37289
  committed
  28a6f68
- nit
  Sh0g0-1758
  committed
  462eb65
- update setup scripts to use the latest release
  anant37289
  committed
  e8f5f29
Commits on Oct 31, 2025
- mat-vec reduction to nodegroup design
  Sh0g0-1758
  committed
  e8e10d3
- data transfer scheme for vec-mat
  Sh0g0-1758
  committed
  cb38a01
- copy to host
  Sh0g0-1758
  committed
  ba21a03
- back to sizes
  Sh0g0-1758
  committed
  92e696f
- fix race condition
  Sh0g0-1758
  committed
  92c3506

Commits

Commits on Sep 23, 2025

Commits on Sep 24, 2025

Commits on Sep 27, 2025

Commits on Sep 29, 2025

Commits on Sep 30, 2025

Commits on Oct 1, 2025

Commits on Oct 2, 2025

Commits on Oct 5, 2025

Commits on Oct 6, 2025

Commits on Oct 8, 2025

Commits on Oct 9, 2025

Commits on Oct 12, 2025

Commits on Oct 13, 2025

Commits on Oct 14, 2025

Commits on Oct 15, 2025

Commits on Oct 17, 2025

Commits on Oct 18, 2025

Commits on Oct 19, 2025

Commits on Oct 22, 2025

Commits on Oct 23, 2025

Commits on Oct 28, 2025

Commits on Oct 29, 2025

Commits on Oct 31, 2025