EfficientPlace: Reinforcement Learning within Tree Search for Fast Macro Placement

This is the code for our paper "Reinforcement Learning within Tree Search for Fast Macro Placement". Zijie Geng, Jie Wang, Ziyan Liu, Siyuan Xu, Zhentao Tang, Mingxuan Yuan, Jianye Hao, Yongdong Zhang, Feng Wu. ICML 2024.

Environment

To build the environment, you can follow commands in scripts/environment.sh.

Usage

To run the code,

python main.py benchmark@_global_=adaptec1

Citation

If you find this code useful, please consider citing the following paper.

@inproceedings{
    geng2024reinforcement,
    title={Reinforcement Learning within Tree Search for Fast Macro Placement},
    author={Zijie Geng and Jie Wang and Ziyan Liu and Siyuan Xu and Zhentao Tang and Mingxuan Yuan and Jianye HAO and Yongdong Zhang and Feng Wu},
    booktitle={Forty-first International Conference on Machine Learning},
    year={2024},
    url={https://openreview.net/forum?id=AJGwSx0RUV}
}