├── .gitignore ├── README.md ├── plot.py ├── plot ├── loss_epoch.py └── overall_loss.py ├── requirements.txt ├── run ├── gen_dataset.sh ├── test.sh ├── test_large.sh ├── train.sh └── train_large.sh ├── src ├── api_deepgate2.py ├── bert_model │ ├── __init__.py │ ├── attention │ │ ├── __init__.py │ │ ├── multi_head.py │ │ └── single.py │ ├── bert.py │ ├── embedding │ │ ├── __init__.py │ │ ├── bert.py │ │ ├── position.py │ │ ├── segment.py │ │ └── token.py │ ├── language_model.py │ ├── transformer.py │ └── utils │ │ ├── __init__.py │ │ ├── feed_forward.py │ │ ├── gelu.py │ │ ├── layer_norm.py │ │ └── sublayer.py ├── config.py ├── convert_aig_to_npz.py ├── dg_datasets │ ├── aig_parser.py │ ├── dataset_utils.py │ ├── dg2_parser.py │ ├── dg3_dataset.py │ ├── dg3_multi_parser.py │ └── dg3_parser.py ├── gen_dataset.py ├── models │ ├── baseline_tf.py │ ├── dg2.py │ ├── dg2_plain.py │ ├── dg2_slice.py │ ├── dg3 copy.py │ ├── dg3.py │ ├── dg_mamba.py │ ├── hop_tf.py │ ├── mha.py │ ├── mlp.py │ ├── path_tf copy.py │ ├── path_tf.py │ ├── plain_tf.py │ ├── pool.py │ └── tf_pool.py ├── simulator │ ├── graph.txt │ ├── out.txt │ ├── simulator │ └── simulator.cpp ├── test_dg3.py ├── test_dg3_large.py ├── train_dg3.py ├── train_dg3_cl.py ├── trainer │ ├── __init__.py │ ├── dg3_trainer copy.py │ ├── dg3_trainer.py │ ├── func_task.py │ ├── optim_schedule.py │ └── pretrain.py └── utils │ ├── circuit_utils.py │ ├── dag_utils.py │ ├── dataset_utils.py │ └── utils.py └── trained ├── dg2_100p.pth ├── log-2023-11-29-13-12.txt ├── model_last.pth └── model_last_workload.pth /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/README.md -------------------------------------------------------------------------------- /plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/plot.py -------------------------------------------------------------------------------- /plot/loss_epoch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/plot/loss_epoch.py -------------------------------------------------------------------------------- /plot/overall_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/plot/overall_loss.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/requirements.txt -------------------------------------------------------------------------------- /run/gen_dataset.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/run/gen_dataset.sh -------------------------------------------------------------------------------- /run/test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/run/test.sh -------------------------------------------------------------------------------- /run/test_large.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/run/test_large.sh -------------------------------------------------------------------------------- /run/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/run/train.sh -------------------------------------------------------------------------------- /run/train_large.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/run/train_large.sh -------------------------------------------------------------------------------- /src/api_deepgate2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/api_deepgate2.py -------------------------------------------------------------------------------- /src/bert_model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/bert_model/__init__.py -------------------------------------------------------------------------------- /src/bert_model/attention/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/bert_model/attention/__init__.py -------------------------------------------------------------------------------- /src/bert_model/attention/multi_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/bert_model/attention/multi_head.py -------------------------------------------------------------------------------- /src/bert_model/attention/single.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/bert_model/attention/single.py -------------------------------------------------------------------------------- /src/bert_model/bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/bert_model/bert.py -------------------------------------------------------------------------------- /src/bert_model/embedding/__init__.py: -------------------------------------------------------------------------------- 1 | from .bert import BERTEmbedding 2 | -------------------------------------------------------------------------------- /src/bert_model/embedding/bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/bert_model/embedding/bert.py -------------------------------------------------------------------------------- /src/bert_model/embedding/position.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/bert_model/embedding/position.py -------------------------------------------------------------------------------- /src/bert_model/embedding/segment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/bert_model/embedding/segment.py -------------------------------------------------------------------------------- /src/bert_model/embedding/token.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/bert_model/embedding/token.py -------------------------------------------------------------------------------- /src/bert_model/language_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/bert_model/language_model.py -------------------------------------------------------------------------------- /src/bert_model/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/bert_model/transformer.py -------------------------------------------------------------------------------- /src/bert_model/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/bert_model/utils/__init__.py -------------------------------------------------------------------------------- /src/bert_model/utils/feed_forward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/bert_model/utils/feed_forward.py -------------------------------------------------------------------------------- /src/bert_model/utils/gelu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/bert_model/utils/gelu.py -------------------------------------------------------------------------------- /src/bert_model/utils/layer_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/bert_model/utils/layer_norm.py -------------------------------------------------------------------------------- /src/bert_model/utils/sublayer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/bert_model/utils/sublayer.py -------------------------------------------------------------------------------- /src/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/config.py -------------------------------------------------------------------------------- /src/convert_aig_to_npz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/convert_aig_to_npz.py -------------------------------------------------------------------------------- /src/dg_datasets/aig_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/dg_datasets/aig_parser.py -------------------------------------------------------------------------------- /src/dg_datasets/dataset_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/dg_datasets/dataset_utils.py -------------------------------------------------------------------------------- /src/dg_datasets/dg2_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/dg_datasets/dg2_parser.py -------------------------------------------------------------------------------- /src/dg_datasets/dg3_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/dg_datasets/dg3_dataset.py -------------------------------------------------------------------------------- /src/dg_datasets/dg3_multi_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/dg_datasets/dg3_multi_parser.py -------------------------------------------------------------------------------- /src/dg_datasets/dg3_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/dg_datasets/dg3_parser.py -------------------------------------------------------------------------------- /src/gen_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/gen_dataset.py -------------------------------------------------------------------------------- /src/models/baseline_tf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/models/baseline_tf.py -------------------------------------------------------------------------------- /src/models/dg2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/models/dg2.py -------------------------------------------------------------------------------- /src/models/dg2_plain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/models/dg2_plain.py -------------------------------------------------------------------------------- /src/models/dg2_slice.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/models/dg2_slice.py -------------------------------------------------------------------------------- /src/models/dg3 copy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/models/dg3 copy.py -------------------------------------------------------------------------------- /src/models/dg3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/models/dg3.py -------------------------------------------------------------------------------- /src/models/dg_mamba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/models/dg_mamba.py -------------------------------------------------------------------------------- /src/models/hop_tf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/models/hop_tf.py -------------------------------------------------------------------------------- /src/models/mha.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/models/mha.py -------------------------------------------------------------------------------- /src/models/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/models/mlp.py -------------------------------------------------------------------------------- /src/models/path_tf copy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/models/path_tf copy.py -------------------------------------------------------------------------------- /src/models/path_tf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/models/path_tf.py -------------------------------------------------------------------------------- /src/models/plain_tf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/models/plain_tf.py -------------------------------------------------------------------------------- /src/models/pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/models/pool.py -------------------------------------------------------------------------------- /src/models/tf_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/models/tf_pool.py -------------------------------------------------------------------------------- /src/simulator/graph.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/simulator/graph.txt -------------------------------------------------------------------------------- /src/simulator/out.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/simulator/out.txt -------------------------------------------------------------------------------- /src/simulator/simulator: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/simulator/simulator -------------------------------------------------------------------------------- /src/simulator/simulator.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/simulator/simulator.cpp -------------------------------------------------------------------------------- /src/test_dg3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/test_dg3.py -------------------------------------------------------------------------------- /src/test_dg3_large.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/test_dg3_large.py -------------------------------------------------------------------------------- /src/train_dg3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/train_dg3.py -------------------------------------------------------------------------------- /src/train_dg3_cl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/train_dg3_cl.py -------------------------------------------------------------------------------- /src/trainer/__init__.py: -------------------------------------------------------------------------------- 1 | from .pretrain import BERTTrainer 2 | -------------------------------------------------------------------------------- /src/trainer/dg3_trainer copy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/trainer/dg3_trainer copy.py -------------------------------------------------------------------------------- /src/trainer/dg3_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/trainer/dg3_trainer.py -------------------------------------------------------------------------------- /src/trainer/func_task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/trainer/func_task.py -------------------------------------------------------------------------------- /src/trainer/optim_schedule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/trainer/optim_schedule.py -------------------------------------------------------------------------------- /src/trainer/pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/trainer/pretrain.py -------------------------------------------------------------------------------- /src/utils/circuit_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/utils/circuit_utils.py -------------------------------------------------------------------------------- /src/utils/dag_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/utils/dag_utils.py -------------------------------------------------------------------------------- /src/utils/dataset_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/utils/dataset_utils.py -------------------------------------------------------------------------------- /src/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/src/utils/utils.py -------------------------------------------------------------------------------- /trained/dg2_100p.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/trained/dg2_100p.pth -------------------------------------------------------------------------------- /trained/log-2023-11-29-13-12.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/trained/log-2023-11-29-13-12.txt -------------------------------------------------------------------------------- /trained/model_last.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/trained/model_last.pth -------------------------------------------------------------------------------- /trained/model_last_workload.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zyzheng17/DeepGate3/HEAD/trained/model_last_workload.pth --------------------------------------------------------------------------------