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.