Skip to content

Conversation

@simonguozirui
Copy link
Collaborator

@simonguozirui simonguozirui commented Feb 23, 2025

Here we investigate the kernels from Sakana's AI CUDA Engineer (blog, paper).

⚠️ Sakana uses the KernelBench Problem, they uses their own eval.

In particular, we focus analysis on 2 problems.

  • Level 1 Problem 15: 15_Matmul_for_lower_triangular_matrices
  • Level 2 Problem 23: 23_Conv3d_GroupNorm_Mean

Thank you to @luchris429 @alexzhang13 @bkal01 for helping with the investigation

Please see more detail in sakana_kernels/README.md

Co-authored-by: bkal01 <bkal01@users.noreply.github.com>
simonguozirui and others added 3 commits February 23, 2025 15:44
Co-authored-by: Bhavesh Kalisetti <bhaveshkalisetti@gmail.com>
Co-authored-by: alexzhang13 <alex.lx.zhang@gmail.com> Bhavesh Kalisetti <bhaveshkalisetti@gmail.com>
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