Download docs/performance.md from HWresearch/GNN4Colliders: direct link, hf CLI and curl.
- Browser
- Download file 1.05 kB
-
https://huggingface.co/HWresearch/GNN4Colliders/resolve/refs%2Fpr%2F18/docs/performance.md
- Command line
-
hf download hf://HWresearch/GNN4Colliders@refs/pr/18/docs/performance.md
-
curl -L -o performance.md https://huggingface.co/HWresearch/GNN4Colliders/resolve/refs%2Fpr%2F18/docs/performance.md
Performance notes
The scripts in ../benchmarks/ separate setup, warmup, and
steady-state timings and emit JSON lines with execution metadata. Float32 eager
execution remains the correctness reference path.
fully_connected_edges has a bounded 32-entry cache keyed by node count,
self-loop policy, and device. It preserves source-major ordering and only
reuses topology indices; event-dependent edge features are always recomputed.
The cache is graph-specific and does not alter scientific behavior.
Training already uses zero_grad(set_to_none=True) and inference already uses
torch.inference_mode() with detached CPU accumulation.
Mixed precision, torch.compile, custom kernels, aggressive worker defaults,
and cache-format replacement were not retained without target-machine
measurements. The main known bottleneck is the quadratic graph workload
N * (N - 1) and associated DGL message passing; size-aware batching and
streaming prediction remain follow-up work because they affect ordering or
output semantics.