Step-TP: A Grounded, Step-Level Dataset with Chain-of-Thought Reasoning for LLM-Guided Tensor Program Optimization
Step-TP: 一个基于步骤级、带有思维链推理的 LLM 引导张量程序优化数据集
机构 * The University of Hong Kong(香港大学) ; University of Science and Technology of China(中国科学技术大学)
AI总结 为解决 LLM 在张量程序优化中缺乏可验证步骤级监督的问题,提出 Step-TP 数据集,通过结构化思维链推理和原子步骤监督实现可靠的多步优化。