Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
ktransformers
Commits
fd8037cd
Commit
fd8037cd
authored
Feb 10, 2025
by
unicornchan
Browse files
Merge branch 'feat-DeepSeekV3' of github.com:kvcache-ai/ktrans_v0.2_dev into feat-DeepSeekV3
parents
c7e6d090
2d684ee9
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
24 additions
and
2 deletions
+24
-2
README.md
README.md
+2
-2
doc/en/DeepseekR1_V3_tutorial.md
doc/en/DeepseekR1_V3_tutorial.md
+22
-0
No files found.
README.md
View file @
fd8037cd
...
...
@@ -43,8 +43,8 @@ https://github.com/user-attachments/assets/ebd70bfa-b2c1-4abb-ae3b-296ed38aa285
-
**[NEW!!!] Local 671B DeepSeek-Coder-V3/R1:**
Running its Q4_K_M version using only 12GB VRAM and 382GB DRAM.
-
Prefill Speed:
-
KTransfermor: 54.21 (32 cores) → 74.362 (dual-socket, 2×32 cores) →
xxx
(optimized AMX-based MoE kernel, v3 only) →
XXX
(selectively using 6 experts, v3 only)
-
Compared to 4.51 tokens/s in llama.cpp with 2×32 cores, achieving up to
**
XXX
× speedup**
.
-
KTransfermor: 54.21 (32 cores) → 74.362 (dual-socket, 2×32 cores) →
255.26
(optimized AMX-based MoE kernel, v3 only) →
286.55
(selectively using 6 experts, v3 only)
-
Compared to 4.51 tokens/s in llama.cpp with 2×32 cores, achieving up to
**
63.53
× speedup**
.
-
Decode Speed(tokens/s):
-
KTransfermor: 8.73 (32 cores) → 11.26 (dual-socket, 2×32 cores) → 13.69 (selectively using 6 experts, v3 only)
-
Compared to 4.51 tokens/s in llama.cpp with 2×32 cores, achieving up to
**3.03× speedup**
.
...
...
doc/en/DeepseekR1_V3_tutorial.md
View file @
fd8037cd
...
...
@@ -24,6 +24,28 @@ gpu: 4090D 24G VRAM <br>
**The highest speedup reaches up to <u>x3.03</u> in decoding and <u>x9.44</u> in prefill.**
### V0.3-Preview
#### settings
-
model: DeepseekV3-BF16 (online quant into int8 for CPU and int4 for GPU)
-
CPU: cpu_model_name:Intel(R) Xeon(R) Gold 6454S, 32 cores per socket, 2 socket, 2 numa nodes
-
GPU: (1~4)x 4090D 24GVRAM (requires more VRAM for longer prompt)
#### memory consumptions:
-
644GB DRAM, at least 12GB VRAM
#### Benchmark Results
| Prompt length | 1K | 2K | 4K | 8K |
|---------------|-----|-----|-----|-----|
| KTrans (8 experts) Prefill token/s | 185.96 | 255.26 | 252.58 | 195.62 |
| KTrans (6 experts) Prefill token/s | 203.70 | 286.55 | 271.08 | 207.20 |
**The prefill of KTrans V0.3 is up to <u>x3.45</u> times faster than KTrans V0.2, and is up to <u>x63.53</u> times faster than Llama.**
**The decoding speed is the same as KTrans V0.2 (6 experts version) so it is omitted.**
The main acceleration comes from
-
Intel AMX instruction set and our specially designed cache friendly memory layout
-
Expert selection strategy that selects fewer experts based on offline profile results of out of domain data
## how to run
### v0.2 showcase
#### single socket version(32 cores)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment