贡献

贡献#

如何为 KernelGenBench 做贡献。

贡献方式#

领域

描述

新算子

用新测试用例扩展基准

新芯片后端

扩展对其他硬件的支持

新智能体

集成 Codex、Trae、Cursor 等编程工具

新方法

贡献专门的优化流程

快速入门#

  1. Fork 仓库

  2. 创建特性分支

  3. 进行修改

  4. 提交拉取请求

开发环境设置#

# Clone your fork
git clone https://github.com/YOUR_USERNAME/KernelGenBench.git
cd KernelGenBench

# Install in development mode
pip install -r requirements/requirements_nvidia.txt
pip install -e .

# Run tests
python scripts/generate_kernel_and_verify.py --single-test --server-type openai

代码风格#

  • Python 代码遵循 PEP 8 规范

  • 在适当的地方使用类型提示

  • 为公共函数添加文档字符串

拉取请求指南#

  1. 每个 PR 一个特性 - 保持修改专注

  2. 添加测试 - 确保新代码经过测试

  3. 更新文档 - 保持文档最新

  4. 关联 issue - 引用相关问题

目录结构#

src/kernelgenbench/    # Core package
├── accuracy/          # Accuracy tests
├── dataset/           # Dataset definitions
└── ...

agent_bench/           # Agent framework
├── methods/           # Agent methods
└── templates/         # Prompt templates

代码审查#

所有提交都需要维护者审查。

许可证#

通过贡献，您同意您的贡献将根据 Apache 2.0 许可证授权。