Skip to content

MusaToTheMoon/rl_pretrain

 
 

Repository files navigation

Workflow

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

About

Exploration of different architectures for the 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

  • Python 100.0%