mlsanityData to 14 Oct 2026

DataKernelBench: Can LLMs Optimize Database Queries on GPUs?

arXiv 2608.25061 ↗First posted 25 Aug 2026cs.CL cs.AI cs.DBcited by 1

Not on the map. Why?

Abstract (arXiv)

GPUs increasingly accelerate database systems, but query-specific peak performance still often relies on hand-written kernels. Existing LLM kernel benchmarks focus on machine learning operators, leaving irregular, heterogeneous, data-movement-heavy database-style operators untested. We introduce DataKernelBench, which translates SQL into validated PyTorch TorchPlan programs and evaluates LLMs that optimize either the core tensor-bounded snippet or the full query in CUDA or Triton through execution-guided repair. Across ten proprietary and open-weight models on TPC-H SF10 with an H100 GPU, the strongest full-query CUDA configuration achieves 2.11× speedup over the TorchPlan baseline at full pass rate. We find that higher-performing implementations commonly use kernel fusion and execution-strategy changes, stronger models benefit most from full-query specialization, and workload context matters more than hardware context. To handle data larger than GPU memory, we extend TorchPlan with Dask-cuDF for on-demand partition loading on TPC-H SF100 with four H100 GPUs, achieving 2.54× speedup. Project page: https://kerneldf.github.io/datakernelbench

Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 22 of its 43 references that are in the map. First posted on arXiv 25 Aug 2026.

Builds on

22 of 43 references.

References · 22

  1. 2022Constitutional AI: Harmlessness from AI Feedbackcited by 3,050
  2. 2024LiveCodeBench: Holistic and Contamination Free Evaluation of Large Language Models for Codecited by 2,004
  3. 2025gpt-oss-120b & gpt-oss-20b Model Cardcited by 1,386
  4. 2026DeepSeek-V4: Towards Highly Efficient Million-Token Context Intelligencecited by 1,040
  5. 2025GLM-4.5: Agentic, Reasoning, and Coding (ARC) Foundation Modelscited by 467
  6. 2023Text-to-SQL Empowered by Large Language Models: A Benchmark Evaluationcited by 385
  7. 2025KernelBench: Can LLMs Write Efficient GPU Kernels?cited by 225
  8. 2017The Case for Learned Index Structurescited by 188
  9. 2019Neo: A Learned Query Optimizercited by 80
  10. 2025Kevin: Multi-Turn RL for Generating CUDA Kernelscited by 78
  11. 2025TritonBench: Benchmarking Large Language Model Capabilities for Generating Triton Operatorscited by 75
  12. 2024Evaluating Language Models for Efficient Code Generationcited by 73
10 more
  1. 2025Towards Robust Agentic CUDA Kernel Benchmarking, Verification, and Optimizationcited by 56
  2. 2026CUDA Agent: Large-Scale Agentic RL for High-Performance CUDA Kernel Generationcited by 47
  3. 2025Devstral: Fine-tuning Language Models for Coding Agent Applicationscited by 24
  4. 2025TritonRL: Training LLMs to Think and Code Triton Without Cheatingcited by 24
  5. 2022Query Processing on Tensor Computation Runtimescited by 12
  6. 2022Share the Tensor Tea: How Databases can Leverage the Machine Learning Ecosystemcited by 7
  7. 2026Bespoke OLAP: Synthesizing Workload-Specific One-size-fits-one Database Enginescited by 7
  8. 2025Rethinking Analytical Processing in the GPU Eracited by 2
  9. 2025GPU Acceleration of SQL Analytics on Compressed Datacited by 1
  10. 2025PystachIO: Efficient Distributed GPU Query Processing with PyTorch over Fast Networks & Fast Storagecited by 1

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 1.

1 paper.