Skip to content

GNN based RL agent for automatic code optimization in the Tiramisu compiler leveraging pretrained access matrices autoencoder.

Notifications You must be signed in to change notification settings

Modern-Compilers-Lab/GNN_RL_Pretrain

Repository files navigation

Workflow

conda env create -f environment.yaml
conda activate tiramisu-build-env
python train_ppo_gnn.py 

About

GNN based RL agent for automatic code optimization in the Tiramisu compiler leveraging pretrained access matrices autoencoder.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages