This repository contains the implementation of Layout Symmetry Annotation for Analog Circuits with GraphNeural Networks, ASPDAC 2021.
Graph structure is widely adopted in EDA. The netlists of analog circuits can be represented with hypergraphs and symmetry constraints of analog circuit layouts can be seen as vertice pair on those graphs. So it is intuitive to introduce recent graph neural network methodology to symmetry constraint detection problem.
This repository implements a GraphSage-based model to predict device-level symmetry constraints for analog circuits. The detailed methodology is described in the paper:
We develop in Python 3.7.5 with below packages:
numpy 1.17.3
networkx 2.4
scipy 1.3.2
matplotlib 3.1.1
torch 1.3.1
torchvision 0.4.2
scikit-learn 0.21.3
We do experiments on extracted leaf-level subcircuits from the two datasets: MAGICAL, ALIGN.
-
prepare the data:
$ bash util/prepare_dataset.sh $ python graphsage/readgraph.py -
train and test:
$ python test.py -
run baseline:
$ python s3det/s3det.py
There are also some options:
- with cuda
- modify hyperparameters
@inproceedings{gao2021layout,
title={Layout symmetry annotation for analog circuits with graph neural networks},
author={Gao, Xiaohan and Deng, Chenhui and Liu, Mingjie and Zhang, Zhiru and Pan, David Z and Lin, Yibo},
booktitle={Proceedings of the 26th Asia and South Pacific Design Automation Conference},
pages={152--157},
year={2021}
}
The orginal version of our project is forked from https://github.com/Chenhui1016/directed_graphsage, which is based on https://github.com/williamleif/graphsage-simple/.