redhat-et / redhat-et/code-agent

Feasibility study into implementing cuda kernels in the current RL framework

Open
#22 0 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Dominant language
Python
Stars
1
Forks
4
PR merge metrics
No merged PRs in 30d

Description

Overview

Investigate using KernelBench in our current RL framework
The objective would be to get a pure cuda kernel C++ implementation of each python based kernel, compile it, verify and then find a baseline using NVIDIA's Nsight's (ncu) tool, once we have the base line we use common best practices of kernel optimization , and feed that as a prompt to a model, then implement the suggested response and repeat the process until we have a really performative kernel (as an example using Elapsed Cycles as the reward)

Contributor guide

No contributing guide indexed for this repository

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start by reviewing the current RL framework and the linked KernelBench project to determine how the Python-based kernels are represented and evaluated. Assess a path for producing, compiling, and verifying equivalent CUDA C++ kernels, then establish an Nsight ncu baseline and define what performance result would count as done.

Written by the indexing model from the issue text.

Assessment

Tech stack
cpp, python
Domain
machine-learning, performance
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Quiet
Clarity
Needs clarification
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.