Skip to content

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

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

No packages published

Languages

  • Python 100.0%