Skip to content

This is a deep-learning based model for Electronic Design Automation(EDA), predicting the congestion location.

Notifications You must be signed in to change notification settings

ycchen218/EDA-Congestion-Prediction

Repository files navigation

EDA Congestion Prediction

Update

  • 2023/08/04

    Add model overview.

  • 2023/06/24

    Add the train.py file.

Introduce

This is a deep-learning-based model used to predict the location of congestion. We achieved an SSIM score of 0.863. During the prediction process, we incorporate the macro region feature, and Rectangular Uniform Wire Density (RUDY)[1], which serves as an early estimation of routing demand after placement. and also consider the RUDY pins, which are calculated based on individual pins and the nets connected to them.

Requirement

  1. python3.8
  2. scipy
  3. matplotlib
  4. numpy
  5. pandas
  6. pytorch 1.12.0
  7. tqdm
  8. pytorch_msssim

Model Overview

image

Train

python train.py

--root_path: The path of the data file
--batch_size: The batch size figure
--num_epochs: The training epochs
--learning_rate: learning rate [0,1]
--weight_path: The path to save the model weight
--fig_path: The path of the figure file

Predict

python predict.py

--data_path: The path of the data file
--fig_save_path: The path you want to save figure
--weight_path: The path of the model weight
--output_path: The path of the predict output with .npy file
--congestion_threshold: congestion_threshold [0,1]
--device: If you want to use gpu type "cuda"

Predict result

  1. Tune your own congestion_threshold, the defalt is 0.5 as shown in following figure.
  2. The output coordinate csv file and image array npy file are in the ./output file.
  3. The model predict cost time is 0.71 ~ 1.8 sec.
  4. Congestion value: $normalize(\frac{overflow}{totaltracks})$.

image

Compare with ground truth

image

Cross validation while evalulate the model

SSIM score: 0.863
by pytorch_msssim.SSIM

Reference

[1] P. Spindler and F. M. Johannes, "Fast and accurate routing demand estimation for efficient routability-driven placement," *Design, Automation & Test in Europe Conference & Exhibition*, pp. 1-6, 2007.

About

This is a deep-learning based model for Electronic Design Automation(EDA), predicting the congestion location.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages