Datasetpublic

Tay-eb/mlir-rl-dataset

MLIR-RL Benchmark Datasets Code: https://github.com/Modern-Compilers-Lab/MLIR-RL Paper: Bendib et al. 2024, A Reinforcement Learning Environment for Automatic Code Optimization in the MLIR Compiler — https://arxiv.org/abs/2409.11068 What is MLIR-RL? MLIR-RL is a reinforcement-learning environment for automatic loop-nest optimization in MLIR. An agent observes linalg operations (matmul, conv, …) and learns sequences of transformations — tiling, interchange… See the full description on the dataset page: https://huggingface.co/datasets/Tay-eb/mlir-rl-dataset.

sourceHugging Faceotherupdated 2d agoView on Hugging Face

0likes63downloads

32 commits on main

917ed142d ago

Update README.md

Tay-eb

c9c4cba2d ago

Mention NN op types (add/matmul/conv/relu/pooling) in mlir_rl_v1_paper row

Tay-eb

b398c7a2d ago

Trim usage/split sections to GitHub ref, drop notes, add citation, fix Code/Paper line break

Tay-eb

b2344752d ago

Update README: MLIR-RL description, GitHub + paper links, subfolder docs

Tay-eb

f41a24c11d ago

Update dataset card for mlir_rl_v1_paper

Tay-eb

c03a82811d ago

Add mlir_rl_v1_paper: merged benchmarks + full models

Tay-eb

8c1170711d ago

Remove legacy_paper (renamed to mlir_rl_v1_paper)

Tay-eb

fb5fd0611d ago