Name		Name	Last commit message	Last commit date
agent		agent
config		config
env_api		env_api
logs		logs
models		models
pretrain		pretrain
utils/dataset_actor		utils/dataset_actor
.gitignore		.gitignore
README.md		README.md
environment.yaml		environment.yaml
evaluate_ppo_gnn.py		evaluate_ppo_gnn.py
main.py		main.py
tiramisu_api_tutorial.py		tiramisu_api_tutorial.py
train_ppo_gnn.py		train_ppo_gnn.py

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.

Custom properties

5 stars

1 watching

Report repository

Releases

Packages

Contributors

Languages