Skip to content

Official open source repository for "A Timing Engine Inspired Graph Neural Network Model for Pre-Routing Slack Prediction" (DAC 2022)

Notifications You must be signed in to change notification settings

ZeayW/TimingPredict

This branch is 1 commit ahead of, 2 commits behind TimingPredict/TimingPredict:master.

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Oct 11, 2022
61879ab · Oct 11, 2022

History

20 Commits
Oct 13, 2021
Apr 28, 2022
Oct 11, 2022
Nov 13, 2021
Oct 21, 2021
Sep 24, 2022
Oct 11, 2022
Nov 23, 2021
Nov 12, 2021
Nov 13, 2021
Mar 2, 2022
Oct 22, 2021
Sep 24, 2022

Repository files navigation

TimingPredict

This is the official open source repository for "A Timing Engine Inspired Graph Neural Network Model for Pre-Routing Slack Prediction" (DAC 2022).

Dataset

Please refer to our Dataset repo for download instructions and documentation.

Usage

First, download the dataset and extract to <repo root>/data/.

Then, download our pre-trained weight here or here or here and put it under <repo root>/checkpoints/08_atcd_specul/15799.pth. (Note that Google drive is showing it as an archive but please download it as a whole.)

Finally, run the following command:

python3 train_gnn.py --test_iter 15799 --checkpoint 08_atcd_specul

This prints two tables indicating the net delay and arrival time prediction results, same as that in our paper.

Code structure

data_graph.py: the data reader and pre-processor.

model.py: our customized GNN model implementation (please refer to the class NetConv, SignalProp, and TimingGCN). also inside it is our GCNII baseline implementation.

train_gnn.py: the model training and testing logic.

train_stat_rf.py, train_deepgnn.py: model training and testing for random forest and deep GCNII baselines.

Reference

Please cite our work if you find our code useful.

@inproceedings{mltimerdac22,
 author = {Guo, Zizheng and Liu, Mingjie and Gu, Jiaqi and Zhang, Shuhan and Pan, David Z. and Lin, Yibo},
 booktitle = {Proceedings of the 59th Annual Design Automation Conference 2022},
 organization = {ACM},
 title = {A Timing Engine Inspired Graph Neural Network Model for Pre-Routing Slack Prediction},
 year = {2022}
}

About

Official open source repository for "A Timing Engine Inspired Graph Neural Network Model for Pre-Routing Slack Prediction" (DAC 2022)

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%