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.