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
ox696c
ktransformers
Commits
8a8311cb
Unverified
Commit
8a8311cb
authored
Mar 17, 2025
by
ZiWei Yuan
Committed by
GitHub
Mar 17, 2025
Browse files
Merge pull request #911 from kvcache-ai/patch_v0.2.3post2
🔧
update multi-gpu-fp8-linear and multi-gpu marlin yaml
parents
0e93a09d
19f058ec
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
3 additions
and
3 deletions
+3
-3
ktransformers/optimize/optimize_rules/DeepSeek-V3-Chat-multi-gpu-fp8-linear-ggml-experts.yaml
...s/DeepSeek-V3-Chat-multi-gpu-fp8-linear-ggml-experts.yaml
+2
-2
ktransformers/optimize/optimize_rules/DeepSeek-V3-Chat-multi-gpu-marlin.yaml
...ize/optimize_rules/DeepSeek-V3-Chat-multi-gpu-marlin.yaml
+1
-1
No files found.
ktransformers/optimize/optimize_rules/DeepSeek-V3-Chat-multi-gpu-fp8-linear-ggml-experts.yaml
View file @
8a8311cb
...
@@ -10,7 +10,7 @@
...
@@ -10,7 +10,7 @@
name
:
"
^model
\\
.layers
\\
.(0|[1-9]|[12][0-9])
\\
."
name
:
"
^model
\\
.layers
\\
.(0|[1-9]|[12][0-9])
\\
."
class
:
ktransformers.models.modeling_deepseek_v3.DeepseekV3RotaryEmbedding
class
:
ktransformers.models.modeling_deepseek_v3.DeepseekV3RotaryEmbedding
replace
:
replace
:
class
:
ktransformers.operators.RoPE.
KMoEGateDeepSeek
V3
class
:
ktransformers.operators.RoPE.
YarnRotaryEmbedding
V3
kwargs
:
kwargs
:
generate_device
:
"
cuda:0"
generate_device
:
"
cuda:0"
prefill_device
:
"
cuda:0"
prefill_device
:
"
cuda:0"
...
@@ -18,7 +18,7 @@
...
@@ -18,7 +18,7 @@
name
:
"
^model
\\
.layers
\\
.([3456][0-9])
\\
."
name
:
"
^model
\\
.layers
\\
.([3456][0-9])
\\
."
class
:
ktransformers.models.modeling_deepseek_v3.DeepseekV3RotaryEmbedding
class
:
ktransformers.models.modeling_deepseek_v3.DeepseekV3RotaryEmbedding
replace
:
replace
:
class
:
ktransformers.operators.RoPE.
KMoEGateDeepSeek
V3
class
:
ktransformers.operators.RoPE.
YarnRotaryEmbedding
V3
kwargs
:
kwargs
:
generate_device
:
"
cuda:1"
generate_device
:
"
cuda:1"
prefill_device
:
"
cuda:1"
prefill_device
:
"
cuda:1"
...
...
ktransformers/optimize/optimize_rules/DeepSeek-V3-Chat-multi-gpu-marlin.yaml
View file @
8a8311cb
...
@@ -10,7 +10,7 @@
...
@@ -10,7 +10,7 @@
name
:
"
^model
\\
.layers
\\
.(0|[1-9]|[12][0-9])
\\
."
name
:
"
^model
\\
.layers
\\
.(0|[1-9]|[12][0-9])
\\
."
class
:
ktransformers.models.modeling_deepseek_v3.DeepseekV3RotaryEmbedding
class
:
ktransformers.models.modeling_deepseek_v3.DeepseekV3RotaryEmbedding
replace
:
replace
:
class
:
ktransformers.operators.RoPE.
KMoEGateDeepSeek
V3
class
:
ktransformers.operators.RoPE.
YarnRotaryEmbedding
V3
kwargs
:
kwargs
:
generate_device
:
"
cuda:0"
generate_device
:
"
cuda:0"
prefill_device
:
"
cuda:0"
prefill_device
:
"
cuda:0"
...
...
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