Skip to content

Add batched syr2k - #3348

Merged
lucbv merged 3 commits into
kokkos:developfrom
yasahi-hpc:add-batched-syr2k
Oct 29, 2026
Merged

lucbv merged 3 commits into
kokkos:developfrom
yasahi-hpc:add-batched-syr2k

Conversation

@yasahi-hpc

Copy link
Copy Markdown
Contributor

This PR adds batched syr2k

  • Serial/Team/TeamVector implementations
  • Corresponding unit-tests

Yuuichi Asahi added 3 commits October 23, 2026 22:04
Signed-off-by: Yuuichi Asahi <y.asahi@nr.titech.ac.jp>
Signed-off-by: Yuuichi Asahi <y.asahi@nr.titech.ac.jp>
Signed-off-by: Yuuichi Asahi <y.asahi@nr.titech.ac.jp>

@lucbv lucbv left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This looks fine although could be optimized maybe?

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

It's a bit annoying that the scaling of c by beta requires a separate loop over all the entries of c before the triple loop over n, k, n

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I agree
I just used the reference blas implementation
If you prefer, I can remove if branches for alpha==0 and beta==0

@lucbv
lucbv merged commit 8c36748 into kokkos:develop Oct 29, 2026
26 of 44 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants