White paper · No. 02

WarpStack-GNN — sweep the whole design space

Method: AI · graph neural network Package: 2nm CoWoS HBM4 + GPU Scope: global exploration Series: WarpStack white papers
Abstract

WarpStack-GNN is a graph neural network trained on WarpStack's own 3D output. It is global: once trained for a design, it answers anywhere in that design's material and thermal space, returning the full warpage map in 19.4 ms — 2,146× faster than solving the same package in 3D. Measured against independent full 3D solves, the whole-map difference is 0.49% of peak-to-peak and the correlation with 3D is 0.99996. It is scored on 512 held-out material and thermal configurations kept entirely out of training. This paper reports the accuracy, the per-design training cost, and the point past which the network is cheaper than solving directly.

Predicted vs solved

The full map, without the solve

A network trained on WarpStack's own 3D output returns the full warpage map without solving. It is global: once trained for a design, it answers anywhere in that design's material and thermal space, in milliseconds. The figures below are measured on the 2nm CoWoS reference design.

Predicted vs solved, on a design point the network never saw

2nm CoWoS package · 56 × 57 mm · active surface
2,146× faster

WarpStack-GNN learns a package’s warpage behaviour from results produced by the detailed 3D method, then returns the full map without solving. Here it is asked for a design point held out of training, and the same point is solved from scratch in 3D for comparison.

Three warpage maps side by side: the full 3D solve reference at 1972.3 µm peak-to-peak, the WarpStack-GNN prediction at 1955.4 µm, and the signed difference between them, which is close to two orders of magnitude smaller than the warpage itself
Detailed 3D · WarpStack-GNN · differenceLeft — the 3D reference, 1972.3 µm peak-to-peak, solved from scratch in 41.5 s. Centre — the GNN prediction, 1955.4 µm, returned in 19.4 ms. Right — the signed difference, spanning 26.5 µm across a surface that bows by almost two millimetres.

AI inference vs numerical solve

Per-design runtime and accuracy · single core
2,146×
MethodRuntimevs 3DError vs 3D
Full 3D solve reference41.54 s1×—
WarpStack-GNN GNN19.4 ms2,146×0.49%
WarpStack-GNN returns a full warpage map in 19.4 ms — 2,146× faster than solving the same package in 3D. Measured against six independent full 3D solves, the whole-map difference is 0.49% of peak-to-peak and the peak-to-peak bow itself lands within 1.89%.

How closely the GNN tracks 3D

512 held-out design points, materials and thermal load varied
0.38% RMSE
QuantityHeld-out setFresh 3D solves
Design points5126
Mean 3D peak-to-peak1,393 µm1,699 µm
Whole-map difference0.38%0.49%
Peak-to-peak error1.41%1.89%
Correlation with 3D0.99960.99996
Every held-out point varies the material set and the thermal load, and the held-out material configurations are kept entirely out of training — so the network is scored on combinations it was never shown.

The design it was measured on

2nm CoWoS HBM4 + GPU stack, imported from IEEE 3Dblox
3D reference
Package56.2 × 57.5 mm
Stacked die columns135
Mechanical layers237
Solid elements45,035
Degrees of freedom169,602
Full 3D solve41.54 s
An 8-stack HBM4 + dual-GPU CoWoS package read straight from its IEEE 3Dblox description — every layer, material, and die placement taken from the source design with no hand re-entry. WarpStack-GNN is trained per design, on warpage produced by the detailed 3D method for that same stack.

What it costs to get there

One-time training, then milliseconds per design point
pays back in 120 points
Training labels (1,792 points)32.2 min
Network training50.8 min
Total one-time cost83.1 min
Per design point afterwards19.4 ms
Break-even vs solving directly120 points
Training data comes from WarpStack’s own 3D formulation, generated far more efficiently than a brute-force sweep of solves — 1,792 labelled warpage fields in 32 minutes rather than the day a design point at a time would cost. Past 120 design points the network is cheaper than solving directly, and a 1,000-point sweep drops from 11.5 hours to 83 minutes.
What to know before you rely on it. A network is trained per design: changing a thickness or a footprint changes the model and requires retraining, and the checkpoint refuses to load against a design it was not built for rather than returning plausible numbers for the wrong stack. Accuracy is reported against WarpStack's own 3D method at a fixed resolution, not against measurement. Generalization to designs outside the one a network was trained on has not yet been measured. The 2nm CoWoS mechanical properties are illustrative values for the material classes involved, not calibrated process data — so absolute warpage in microns is indicative; the speedup and the method-to-method agreement do not depend on it. For warpage checked across resolutions, see white paper No. 01.
More in this series

Put warpage in the loop.

WarpStack is in active development. Request a demo or a walkthrough on your own 2.5D/3D chiplet designs, and we'll get you set up.

noveetyai@noveetymanagement.com