Skip to content

Add rolled rotate-and-reduce kernel#2699

Merged
copybara-service[bot] merged 1 commit intogoogle:mainfrom
j2kun:arithmetic-dag-kernel-impl
Feb 27, 2026
Merged

Add rolled rotate-and-reduce kernel#2699
copybara-service[bot] merged 1 commit intogoogle:mainfrom
j2kun:arithmetic-dag-kernel-impl

Conversation

@j2kun
Copy link
Copy Markdown
Collaborator

@j2kun j2kun commented Feb 25, 2026

This PR adds a new unroll API option to the KernelImplementation.h rotate and reduce kernels. Doesn't yet add a pass option to enable it due to the lack of rotation key analysis.

As per discussion with @asraa, I'm leaving out the zeroDiagonals support until a followup PR

@j2kun j2kun force-pushed the arithmetic-dag-kernel-impl branch 7 times, most recently from 2ae9a3b to c131524 Compare February 26, 2026 19:22
@j2kun j2kun changed the title WIP: Add rolled rotate-and-reduce kernel Add rolled rotate-and-reduce kernel Feb 26, 2026
@j2kun j2kun force-pushed the arithmetic-dag-kernel-impl branch from c131524 to 31837a7 Compare February 26, 2026 21:09
@j2kun j2kun added the pull_ready Indicates whether a PR is ready to pull. The copybara worker will import for internal testing label Feb 26, 2026
@j2kun
Copy link
Copy Markdown
Collaborator Author

j2kun commented Feb 26, 2026

Marking as pull_ready so I can patch internal work over this PR

@j2kun j2kun force-pushed the arithmetic-dag-kernel-impl branch from 31837a7 to 8a8fccb Compare February 27, 2026 16:52
@j2kun j2kun removed the pull_ready Indicates whether a PR is ready to pull. The copybara worker will import for internal testing label Feb 27, 2026
@j2kun j2kun requested a review from asraa February 27, 2026 17:34
Comment thread lib/Kernel/TestingUtils.h Outdated
Comment thread lib/Kernel/KernelImplementation.h
@j2kun j2kun force-pushed the arithmetic-dag-kernel-impl branch from 8a8fccb to 3ed180a Compare February 27, 2026 22:19
@j2kun j2kun added the pull_ready Indicates whether a PR is ready to pull. The copybara worker will import for internal testing label Feb 27, 2026
@copybara-service copybara-service Bot merged commit 3970331 into google:main Feb 27, 2026
8 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

pull_ready Indicates whether a PR is ready to pull. The copybara worker will import for internal testing

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants