task-constructor

What problem does it solve?

从 PyTorch/Triton 代码仓中提取算子实现，构建为 KernelBench 格式的标准化单文件自包含任务。支持代码提取、AST 依赖追踪、函数内联、import 清理、格式验证和参考对比测试。当用户需要从现有代码构建 task_code 时使用此 Skill。

Core Features & Use Cases

• 提取算子实现并打包为 KernelBench 单文件任务
• 支持 AST 依赖追踪、函数内联、import 清理、格式验证和参考对比测试
• 场景：从现有代码构建 task_code、包装代码片段生成 KernelBench 任务等

Quick Start

将代码仓路径提供给 task-constructor 并运行以生成一个自包含的 KernelBench 任务文件。