willychan21/

ParallelKernelBench_Problems

Reference problems for ParallelKernelBench: a benchmark for LLM-generated multi-GPU CUDA kernels. This dataset contains 87 reference implementations in reference/ and the input tensor specification in utils/input_output_tensors.py.

	Files




	

Path Description

data/problems.parquet One row per problem (tabular access)

reference/*.py Reference solution() implementations

utils/input_output_tensors.py Input/output tensor… See the full description on the dataset page: https://huggingface.co/datasets/willychan21/ParallelKernelBench_Problems.