xuzhuo0417/openacm-gnn-8bit

OpenACM 8-bit PEA-GNN Surrogate

This repository packages the 8-bit OpenACM PEA-GNN surrogate model files for Hugging Face.

The model predicts five approximate multiplier metrics from graph-structured multiplier configurations:

• mred
• nmed
• delay
• area
• power

Provenance

• OpenACM source project: https://github.com/ShenShan123/OpenACM
• OpenACM source directory: openacm/multiplier_gnn/8-bitGNN
• Helper code, FEATURE.csv, and label_minmax_8.txt come from the OpenACM model files.
• Weight file: best_model_weights_8.pth
• Weight source: new user-provided 8-bit final model weight packaged from the HF dataset model directory

Helper code, FEATURE, and minmax follow the OpenACM GitHub 8-bit model files; the 8-bit weight is the newer user-provided final weight.

Files

• best_model_weights_8.pth: final model parameters.
• gnn_predictor.py: GraphSAGE model definition and prediction wrapper.
• my_io.py: helper code for building PyTorch Geometric graph inputs from multiplier configurations.
• FEATURE.csv: static feature table used by the input builder.
• label_minmax_8.txt: min/max values used to map normalized model outputs to raw target units.
• config.json: compact metadata and hash information.
• requirements.txt: Python package dependencies.

Associated Dataset

https://huggingface.co/datasets/xuzhuo0417/openacm-gnn-8bit-dataset

Download Example

from huggingface_hub import hf_hub_download

weights_path = hf_hub_download(
    repo_id="xuzhuo0417/openacm-gnn-8bit",
    filename="best_model_weights_8.pth"
)
print(weights_path)

Notes

These are final model parameters only, without optimizer or epoch state.