CUDA-Agent-Ops-6K هي مجموعة بيانات تدريبية موجهة لتوليد وتحسين نوى CUDA، أصدرتها BytedTsinghua-SIA في عام 2026، وتهدف إلى توفير مهام تدريبية عالية الجودة على مستوى العمليات الحسابية لتعلم التعزيز للوكلاء الذكيين وتوليد كود CUDA.

تحتوي مجموعة البيانات على 6,000 مهمة تدريبية لمجموعات عمليات حسابية اصطناعية، مستمدة بشكل رئيسي من العمليات الحسابية في مكتبتَي PyTorch وTransformers، حيث يتم دمج عدة عمليات حسابية في مهام اندماج أكثر تعقيدًا باستخدام نماذج اللغة الكبيرة (LLM). تُخزن البيانات بصيغة Parquet، وتُستخدم بشكل أساسي لتوليد نوى CUDA وتحسينها، وكذلك لتدريب الوكلاء الذكيين عبر التعلم المعزز.

تكوين مجموعة البيانات

• ops: قائمة العمليات الحسابية أو الوحدات المتضمنة في المهمة؛
• data_source: معرف مصدر البيانات؛
• code: كود Python/PyTorch قابل للتنفيذ للمهمة.

الاستشهاد

@misc{cuda_agent_2026,
  title        = {CUDA Agent: Large-Scale Agentic RL for High-Performance CUDA Kernel Generation},
  author       = {Dai, Weinan and Wu, Hanlin and Yu, Qiying and Gao, Huan-ang and Li, Jiahao and Jiang, Chengquan and Lou, Weiqiang and Song, Yufan and Yu, Hongli and Chen, Jiaze and Ma, Wei-Ying and Zhang, Ya-Qin and Liu, Jingjing and Wang, Mingxuan and Liu, Xin and Zhou, Hao},
  year         = {2026},
  howpublished = {Project page and technical report}
}

@misc{cuda_agent_ops_6k_2026,
  title        = {CUDA-Agent-Ops-6K: Training Dataset for CUDA-Agent},
  author       = {{BytedTsinghua-SIA}},
  year         = {2026},
  howpublished = {Hugging Face dataset},
  url          = {https://huggingface.co/datasets/BytedTsinghua-SIA/CUDA-Agent-Ops-6K}
}