Commit Graph

1 Commits

Author SHA1 Message Date
7a95983e0f Rewrite single_shot: 8-GPU pipeline parallel
- Loads all 95 shards, assigns layers round-robin across 8 B200s
- ~8 layers per GPU, ~118GB weights per GPU (fits in 183GB)
- 3-phase pipeline: load weights → JIT compile → inference
- Activations move between GPUs at layer boundaries (NVLink)
- No streaming, no shard caching, no per-layer CPU loads
- Includes timing for each phase
2026-05-30 23:36:14 +00:00