Skip to content

Initial Kernel Bench "compiler"#115

Merged
rengolin merged 13 commits intollvm:mainfrom
rengolin:kb
Apr 9, 2026
Merged

Initial Kernel Bench "compiler"#115
rengolin merged 13 commits intollvm:mainfrom
rengolin:kb

Conversation

@rengolin
Copy link
Copy Markdown
Member

@rengolin rengolin commented Apr 8, 2026

New tool to import, compile and run Kernel Bench examples.

So far:

  • Using the new transform to fix args.
  • Overriding Kernel Bench input argument shapes.
  • Default pipeline lowers to loops / scalar code

Next:

  • Start working on an optimization pipeline for CPU
  • Work on GPU execution
  • Work on GPU optimization pipeline

Also:

  • Move test descriptor files to a central location, and add search include path like python schedules

using the new transform to fix args

output is still not correct, but everything else seems to be.
@rengolin rengolin requested review from adam-smnk and tkarna April 8, 2026 17:22
@rengolin rengolin marked this pull request as ready for review April 9, 2026 10:58
Comment thread tools/kernel_bench
Comment thread tools/kernel_bench
Comment thread tools/kernel_bench
Comment thread examples/end-to-end/KernelBench/test_kernel_bench.py
Comment thread examples/end-to-end/KernelBench/test.py Outdated
Comment thread examples/end-to-end/KernelBench/test.py Outdated
Comment thread examples/end-to-end/KernelBench/test_kernel_bench.py
Comment thread tools/kernel_bench Outdated
Comment thread tools/kernel_bench Outdated
Comment thread tools/kernel_bench Outdated
@rengolin rengolin merged commit 32cd309 into llvm:main Apr 9, 2026
3 checks passed
@rengolin rengolin deleted the kb branch April 9, 2026 13:43
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants