abetlen / abetlen/llama-cpp-python

Mixtral A100/H100 performance

Open
#1,017 3 comments 1 reaction 0 assignees View on GitHub
bug performance
Dominant language
Python
Stars
10.6k
Forks
1.4k
PR merge metrics
PR metrics pending

Description

I tried to play with mixtral Q5_K_M quant both on llama.cpp and python. Both servers use cuda 12, but same on 11.6. Here are some results:

llama cpp, A100
```
llama_print_timings: load time = 5438.24 ms
llama_print_timings: sample time = 92.38 ms / 243 runs ( 0.38 ms per token, 2630.35 tokens per second)
llama_print_timings: prompt eval time = 105.61 ms / 9 tokens ( 11.73 ms per token, 85.22 tokens per second)
llama_print_timings: eval time = 5031.24 ms / 242 runs ( 20.79 ms per token, 48.10 tokens per second)
llama_print_timings: total time = 5324.31 ms
```

llama cpp, H100
```
llama_print_timings: load time = 4422.18 ms
llama_print_timings: sample time = 28.22 ms / 190 runs ( 0.15 ms per token, 6732.81 tokens per second)
llama_print_timings: prompt eval time = 77.82 ms / 9 tokens ( 8.65 ms per token, 115.65 tokens per second)
llama_print_timings: eval time = 3114.78 ms / 189 runs ( 16.48 ms per token, 60.68 tokens per second)
llama_print_timings: total time = 3272.63 ms
```

llama-cpp-python, A100

```
llama_print_timings: load time = 3054.10 ms
llama_print_timings: sample time = 113.27 ms / 269 runs ( 0.42 ms per token, 2374.92 tokens per second)
llama_print_timings: prompt eval time = 3053.75 ms / 243 tokens ( 12.57 ms per token, 79.57 tokens per second)
llama_print_timings: eval time = 11053.97 ms / 268 runs ( 41.25 ms per token, 24.24 tokens per second)
llama_print_timings: total time = 14967.72 ms
```

llama-cpp-python, H100
```

llama_print_timings: load time = 2572.72 ms
llama_print_timings: sample time = 44.71 ms / 269 runs ( 0.17 ms per token, 6016.15 tokens per second)
llama_print_timings: prompt eval time = 2572.38 ms / 243 tokens ( 10.59 ms per token, 94.47 tokens per second)
llama_print_timings: eval time = 7424.27 ms / 268 runs ( 27.70 ms per token, 36.10 tokens per second)
llama_print_timings: total time = 10591.14 ms
```

So every time I can see a 10-50% performance drop on the same settings both prompt eval time and generation. Is this behavior expected?

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.