Too slow in 2 x H100, got decode 40t/s

#60
by tientitmu - opened

I installed both the unslothai/llama.cpp and the official version. In all two versions, the prefill got 100t/s at the first glance (speedup after) but decode phase just got 40t/s and two H100 ultilizing only 30%.

Then I checked top, I realized that llama-server just use 1 core (but params are -t 224 -tb 224). So there are something wrong here?

Use a sglang recipe. I got 8k-10k prefill and 50-80 tps decode on a rented rtx 6000 pro at 200k context

I installed both the unslothai/llama.cpp and the official version. In all two versions, the prefill got 100t/s at the first glance (speedup after) but decode phase just got 40t/s and two H100 ultilizing only 30%.

Then I checked top, I realized that llama-server just use 1 core (but params are -t 224 -tb 224). So there are something wrong here?

Hi! Did you try to raise batch-size and ubatch-size? I'm using a 5070Ti and a 3090, and I settled around 2048-2560 but you could go much higher.

Read author's instruction in mtp directory.
I got 60 tok/s w/o mtp and 75-80 tok/s with mtp for same hw.

Sign up or log in to comment