strata_bench.py records a layer-split run as single-GPU (card list and VRAM total)

Chiusa Adatta ai principianti
  • Qualcosa non va?
#2,887 0 commenti 0 reazioni 0 assegnatari Vedi su GitHub

I maintainer di solito rispondono entro 3 giorni

Nessuno ha ancora preso questa issue.

Valutazione

Difficoltà
2/5
Tempo stimato
1-3 ore
Idoneità per principianti
91/100
Tipo di issue
Bug
Chiarezza
Specificata chiaramente
Stato di attività
Attiva
Stack tecnologico
python
Ambito
devtools, testing

Direzione di ricerca

Inizia da tools/strata_bench.py:614-616 ed esamina come viene confrontato static["gpu_name"] con le schede utilizzate localmente. Riproduci con la configurazione di suddivisione dei layer su due GPU descritta e ./bench.sh --url http://127.0.0.1:<port>. Done significa che il README.md e il results.json generati mantengono entrambi i nomi delle GPU servite e riportano i 40952 MiB combinati, mentre i casi con una singola scheda restano invariati.

Scritto dal modello di indicizzazione a partire dal testo della issue.

Descrizione

What happened

A benchmark of a two-card layer split (RTX 3090 + RTX 4080 SUPER) produces a report naming only the first card:

"gpu": ["NVIDIA GeForce RTX 3090"], "gpu_vram_mib": 24576

while the server reports both correctly:

"machine": {"gpu": {"name": "NVIDIA GeForce RTX 3090 + NVIDIA GeForce RTX 4080 SUPER",
                    "total_mib": 40952}}

and the engine log says layer split: layers 0-28 (CUDA0), 29-47 (CUDA1), one hand-off per window.

Cause, tools/strata_bench.py:614-616:

elif len(used) > 1 and static.get("gpu_name"):
    named = [g for g in used if g["name"] == static["gpu_name"]]
    used = named[:1] or used[:1]

With a split, static["gpu_name"] is "A + B", so no single card equals it, named is empty, and used is truncated to the first card. gpu_vram_mib is then summed over that one card (lines 627-628), overriding the server's correct total. The " + " form is already parsed 20 lines below (#2001), but that branch only fires when the served cards are not all local (the other-vendor case), so a split across two local NVIDIA cards never reaches it.

The branch is only reached when cfg is None, i.e. with --url. On current main that is the only way to benchmark at all, because bench.sh crashes when it starts the server itself (#2872). So at the moment every layer-split run benchmarked with the tool is reported as single-GPU, in both the card list and the VRAM total.

To reproduce: a config with "gpu": [0, 1], start the server, then ./bench.sh --url http://127.0.0.1:<port>. The generated README.md and results.json name one card.

Suggested fix: match each name the server gives, keeping the count.

elif len(used) > 1 and static.get("gpu_name"):
    served_names = [n for n in static["gpu_name"].split(" + ") if n and n != "?"]
    named = [g for g in used if g["name"] in served_names]
    used = named[:len(served_names)] or used[:1]

Checked against this machine's values: the split case becomes ["RTX 3090", "RTX 4080 SUPER"] / 40952 MiB, and both single-card-of-two cases are unchanged.

Strata version, GPU, OS

Engine 0.1.45, main at 477ac13c (these lines unchanged on main today). RTX 3090 + RTX 4080 SUPER, Linux (Ubuntu 24.04 base), engine compiled from source.

Lingua principale
C++
Stelle
11.6k
Fork
1k
Nessuna PR unita negli ultimi 30g

Preparare l'ambiente

Questo progetto non fornisce container di sviluppo, Dockerfile né guida per i contributori, quindi l'ambiente è a tuo carico: parti dal suo README e consulta la nostra guida al primo contributo per i passaggi generali.

Come iniziare

  1. Leggi tutta la issue e poi la guida ai contributi del progetto.
  2. Commenta sulla issue per dire che te ne occupi tu — evita che due persone facciano lo stesso lavoro.
  3. Fai un fork del repository e lavora su un branch.
  4. Apri una pull request che faccia riferimento al numero della issue.

Altre issue di Niko1221/Strata

Tutte le issue di Niko1221/Strata

Issue simili

Altre issue su C++

Ricevi le nuove issue nella tua casella

Un breve riepilogo di issue GitHub adatte ai principianti.