Skip to content

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

 
 
 

Contributors

Languages