研究人员开发了DataKernelBench,这是一个旨在评估大型语言模型(LLM)针对GPU优化数据库查询能力的新基准。与专注于机器学习算子的现有基准不同,DataKernelBench解决了数据密集型数据库操作的独特挑战。该基准将SQL查询转换为PyTorch TorchPlan程序,使LLM能够使用CUDA或Triton优化特定的代码片段或整个查询。使用NVIDIA H100 GPU在TPC-H数据集上进行的初步测试表明,LLM优化后的查询比基线实现快2.11倍,而更强大的模型在完全查询专业化方面显示出更大的优势。 AI
影响 该基准可能会推动更适合优化复杂数据库操作的LLM的发展,从而可能加速AI驱动应用程序中的数据处理。
排序理由 该集群包含一篇研究论文,介绍了一个用于评估LLM在特定技术任务上性能的新基准。[lever_c_demoted from research: ic=1 ai=1.0]
- CUDA
- Dask-cuDF
- DataKernelBench
- graphics processing unit
- LLMs
- NVIDIA H100
- PyTorch
- TorchPlan
- TPC-H
- Triton
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →