code-diff
RS-DiffusionCode_Opt_Triton
Overview
This dataset, TEEN-D/Code_Opt_Triton, is an extended version of the publicly available GPUMODE/Inductor_Created_Data_Permissive dataset. It contains pairs of original (PyTorch or Triton) programs and their equivalent Triton code (generated by torch inductor), intended for training models in PyTorch-to-Triton code translation and optimization.
The primary modification in this extended version is that each optimized Triton code snippet is paired with both its original source… See the full description on the dataset page: https://huggingface.co/datasets/Teen-Different/Code_Opt_Triton.diffusion-easy-hard-repro-codefull-repo-coverage-diffCode_Opt_Triton_Shuffled
TEEN-D/Code_Opt_Triton_Shuffled
Overview
This dataset, TEEN-D/Code_Opt_Triton_Shuffled, is a shuffled version of the extended TEEN-D/Code_Opt_Triton dataset (which itself is an extension of GPUMODE/Inductor_Created_Data_Permissive). It provides a collection of pairs of original (PyTorch or Triton) programs and their corresponding optimized Triton code, designed for training machine learning models for code translation and optimization tasks targeting GPUs.
The key… See the full description on the dataset page: https://huggingface.co/datasets/Teen-Different/Code_Opt_Triton_Shuffled.train_adv_mean_diff_code_sabotage_safe