FazBrowse GitHub Viewer
|
Trending
|
URL:
|
Home
Tools:
[Download Repo ZIP]
[View Raw Code]
[Original HTTPS Page]
llama.cpp/src/llama-model.cpp at dflash-docs · kashif/llama.cpp · GitHub
kashif
/
llama.cpp
Public
forked from
ggml-org/llama.cpp
Notifications
You must be signed in to change notification settings
Fork
0
Star
0
Code
Pull requests
0
Actions
Projects
Security and quality
0
Insights
Additional navigation options
Code
Pull requests
Actions
Projects
Security and quality
Insights
Expand file tree
Breadcrumbs
llama.cpp
/
src
/
llama-model.cpp
Copy path
More file actions
More file actions
Latest commit
History
History
History
2717 lines (2419 loc) · 120 KB
Breadcrumbs
llama.cpp
/
src
/
llama-model.cpp
Copy path
File metadata and controls
2717 lines (2419 loc) · 120 KB
Raw
Copy raw file
Download raw file
Open symbols panel
Edit and raw actions
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
977
978
979
980
981
982
983
984
985
986
987
988
989
990
991
992
993
994
995
996
997
998
999
1000
#
include
"
llama-model.h
"
#
include
"
llama-arch.h
"
#
include
"
llama-ext.h
"
#
include
"
llama-hparams.h
"
#
include
"
llama-impl.h
"
#
include
"
llama-mmap.h
"
#
include
"
llama-cparams.h
"
#
include
"
llama-model-loader.h
"
#
include
"
llama-kv-cache.h
"
#
include
"
llama-kv-cache-iswa.h
"
#
include
"
llama-kv-cache-dsa.h
"
#
include
"
llama-memory-hybrid.h
"
#
include
"
llama-memory-hybrid-iswa.h
"
#
include
"
llama-memory-recurrent.h
"
#
include
"
models/models.h
"
#
include
"
ggml.h
"
#
include
"
ggml-cpp.h
"
#
include
<
algorithm
>
#
include
<
cassert
>
#
include
<
cfloat
>
#
include
<
cstdint
>
#
include
<
cstring
>
#
include
<
cmath
>
#
include
<
functional
>
#
include
<
map
>
#
include
<
numeric
>
#
include
<
regex
>
#
include
<
sstream
>
#
include
<
stdexcept
>
#
include
<
string
>
#
include
<
vector
>
static
llama_model *
llama_model_mapping
(llm_arch arch,
const
llama_model_params & params) {
switch
(arch) {
case
LLM_ARCH_LLAMA
:
return
new
llama_model_llama
(params);
case
LLM_ARCH_LLAMA4
:
return
new
llama_model_llama4
(params);
case
LLM_ARCH_LLAMA_EMBED
:
return
new
llama_model_llama_embed
(params);
case
LLM_ARCH_MAINCODER
:
return
new
llama_model_maincoder
(params);
case
LLM_ARCH_TALKIE
:
return
new
llama_model_talkie
(params);
case
LLM_ARCH_DECI
:
return
new
llama_model_deci
(params);
case
LLM_ARCH_BAICHUAN
:
return
new
llama_model_baichuan
(params);
case
LLM_ARCH_FALCON
:
return
new
llama_model_falcon
(params);
case
LLM_ARCH_GROK
:
return
new
llama_model_grok
(params);
case
LLM_ARCH_STARCODER
:
return
new
llama_model_starcoder
(params);
case
LLM_ARCH_REFACT
:
return
new
llama_model_refact
(params);
case
LLM_ARCH_BERT
:
return
new
llama_model_bert
(params);
case
LLM_ARCH_JINA_BERT_V2
:
return
new
llama_model_jina_bert_v2
(params);
case
LLM_ARCH_JINA_BERT_V3
:
return
new
llama_model_jina_bert_v3
(params);
case
LLM_ARCH_NOMIC_BERT
:
return
new
llama_model_nomic_bert
(params);
case
LLM_ARCH_NOMIC_BERT_MOE
:
return
new
llama_model_nomic_bert_moe
(params);
case
LLM_ARCH_MODERN_BERT
:
return
new
llama_model_modern_bert
(params);
case
LLM_ARCH_NEO_BERT
:
return
new
llama_model_neo_bert
(params);
case
LLM_ARCH_EUROBERT
:
return
new
llama_model_eurobert
(params);
case
LLM_ARCH_BLOOM
:
return
new
llama_model_bloom
(params);
case
LLM_ARCH_MPT
:
return
new
llama_model_mpt
(params);
case
LLM_ARCH_STABLELM
:
return
new
llama_model_stablelm
(params);
case
LLM_ARCH_MELLUM
:
return
new
llama_model_mellum
(params);
case
LLM_ARCH_QWEN
:
return
new
llama_model_qwen
(params);
case
LLM_ARCH_QWEN2
:
return
new
llama_model_qwen2
(params);
case
LLM_ARCH_DREAM
:
return
new
llama_model_dream
(params);
case
LLM_ARCH_LLADA
:
return
new
llama_model_llada
(params);
case
LLM_ARCH_LLADA_MOE
:
return
new
llama_model_llada_moe
(params);
case
LLM_ARCH_RND1
:
return
new
llama_model_rnd1
(params);
case
LLM_ARCH_QWEN2VL
:
return
new
llama_model_qwen2vl
(params);
case
LLM_ARCH_QWEN2MOE
:
return
new
llama_model_qwen2moe
(params);
case
LLM_ARCH_QWEN3
:
return
new
llama_model_qwen3
(params);
case
LLM_ARCH_QWEN3MOE
:
return
new
llama_model_qwen3moe
(params);
case
LLM_ARCH_QWEN3VL
:
return
new
llama_model_qwen3vl
(params);
case
LLM_ARCH_QWEN3VLMOE
:
return
new
llama_model_qwen3vlmoe
(params);
case
LLM_ARCH_PHI2
:
return
new
llama_model_phi2
(params);
case
LLM_ARCH_PHI3
:
return
new
llama_model_phi3
(params);
case
LLM_ARCH_PHIMOE
:
return
new
llama_model_phimoe
(params);
case
LLM_ARCH_PLAMO
:
return
new
llama_model_plamo
(params);
case
LLM_ARCH_PLAMO2
:
return
new
llama_model_plamo2
(params);
case
LLM_ARCH_PLAMO3
:
return
new
llama_model_plamo3
(params);
case
LLM_ARCH_GPT2
:
return
new
llama_model_gpt2
(params);
case
LLM_ARCH_CODESHELL
:
return
new
llama_model_codeshell
(params);
case
LLM_ARCH_ORION
:
return
new
llama_model_orion
(params);
case
LLM_ARCH_INTERNLM2
:
return
new
llama_model_internlm2
(params);
case
LLM_ARCH_MINICPM3
:
return
new
llama_model_minicpm3
(params);
case
LLM_ARCH_GEMMA
:
return
new
llama_model_gemma
(params);
case
LLM_ARCH_GEMMA2
:
return
new
llama_model_gemma2
(params);
case
LLM_ARCH_GEMMA3
:
return
new
llama_model_gemma3
(params);
case
LLM_ARCH_GEMMA3N
:
return
new
llama_model_gemma3n
(params);
case
LLM_ARCH_GEMMA4
:
return
new
llama_model_gemma4
(params);
case
LLM_ARCH_GEMMA4_ASSISTANT
:
return
new
llama_model_gemma4_assistant
(params);
case
LLM_ARCH_GEMMA_EMBEDDING
:
return
new
llama_model_gemma_embedding
(params);
case
LLM_ARCH_STARCODER2
:
return
new
llama_model_starcoder2
(params);
case
LLM_ARCH_MAMBA
:
return
new
llama_model_mamba
(params);
case
LLM_ARCH_MAMBA2
:
return
new
llama_model_mamba2
(params);
case
LLM_ARCH_JAMBA
:
return
new
llama_model_jamba
(params);
case
LLM_ARCH_XVERSE
:
return
new
llama_model_xverse
(params);
case
LLM_ARCH_COMMAND_R
:
return
new
llama_model_command_r
(params);
case
LLM_ARCH_COHERE2
:
return
new
llama_model_cohere2
(params);
case
LLM_ARCH_COHERE2MOE
:
return
new
llama_model_cohere2moe
(params);
case
LLM_ARCH_DBRX
:
return
new
llama_model_dbrx
(params);
case
LLM_ARCH_OLMO
:
return
new
llama_model_olmo
(params);
case
LLM_ARCH_OLMO2
:
return
new
llama_model_olmo2
(params);
case
LLM_ARCH_OLMOE
:
return
new
llama_model_olmoe
(params);
case
LLM_ARCH_OPENELM
:
return
new
llama_model_openelm
(params);
case
LLM_ARCH_GPTNEOX
:
return
new
llama_model_gptneox
(params);
case
LLM_ARCH_ARCTIC
:
return
new
llama_model_arctic
(params);
case
LLM_ARCH_DEEPSEEK
:
return
new
llama_model_deepseek
(params);
case
LLM_ARCH_DEEPSEEK2
:
return
new
llama_model_deepseek2
(params);
case
LLM_ARCH_DEEPSEEK2OCR
:
return
new
llama_model_deepseek2ocr
(params);
case
LLM_ARCH_DEEPSEEK32
:
return
new
llama_model_deepseek32
(params);
case
LLM_ARCH_GLM_DSA
:
return
new
llama_model_glm_dsa
(params);
case
LLM_ARCH_MISTRAL4
:
return
new
llama_model_mistral4
(params);
case
LLM_ARCH_CHATGLM
:
return
new
llama_model_chatglm
(params);
case
LLM_ARCH_GLM4
:
return
new
llama_model_glm4
(params);
case
LLM_ARCH_GLM4_MOE
:
return
new
llama_model_glm4_moe
(params);
case
LLM_ARCH_BITNET
:
return
new
llama_model_bitnet
(params);
case
LLM_ARCH_T5
:
return
new
llama_model_t5
(params);
case
LLM_ARCH_T5ENCODER
:
return
new
llama_model_t5encoder
(params);
case
LLM_ARCH_JAIS
:
return
new
llama_model_jais
(params);
case
LLM_ARCH_JAIS2
:
return
new
llama_model_jais2
(params);
case
LLM_ARCH_NEMOTRON
:
return
new
llama_model_nemotron
(params);
case
LLM_ARCH_NEMOTRON_H
:
return
new
llama_model_nemotron_h
(params);
case
LLM_ARCH_NEMOTRON_H_MOE
:
return
new
llama_model_nemotron_h_moe
(params);
case
LLM_ARCH_EXAONE
:
return
new
llama_model_exaone
(params);
case
LLM_ARCH_EXAONE4
:
return
new
llama_model_exaone4
(params);
case
LLM_ARCH_EXAONE_MOE
:
return
new
llama_model_exaone_moe
(params);
case
LLM_ARCH_RWKV6
:
return
new
llama_model_rwkv6
(params);
case
LLM_ARCH_RWKV6QWEN2
:
return
new
llama_model_rwkv6qwen2
(params);
case
LLM_ARCH_RWKV7
:
return
new
llama_model_rwkv7
(params);
case
LLM_ARCH_ARWKV7
:
return
new
llama_model_arwkv7
(params);
case
LLM_ARCH_GRANITE
:
return
new
llama_model_granite
(params);
case
LLM_ARCH_GRANITE_MOE
:
return
new
llama_model_granite_moe
(params);
case
LLM_ARCH_MINICPM
:
return
new
llama_model_minicpm
(params);
case
LLM_ARCH_GRANITE_HYBRID
:
return
new
llama_model_granite_hybrid
(params);
case
LLM_ARCH_CHAMELEON
:
return
new
llama_model_chameleon
(params);
case
LLM_ARCH_WAVTOKENIZER_DEC
:
return
new
llama_model_wavtokenizer_dec
(params);
case
LLM_ARCH_PLM
:
return
new
llama_model_plm
(params);
case
LLM_ARCH_BAILINGMOE
:
return
new
llama_model_bailingmoe
(params);
case
LLM_ARCH_BAILINGMOE2
:
return
new
llama_model_bailingmoe2
(params);
case
LLM_ARCH_SEED_OSS
:
return
new
llama_model_seed_oss
(params);
case
LLM_ARCH_DOTS1
:
return
new
llama_model_dots1
(params);
case
LLM_ARCH_ARCEE
:
return
new
llama_model_arcee
(params);
case
LLM_ARCH_AFMOE
:
return
new
llama_model_afmoe
(params);
case
LLM_ARCH_ERNIE4_5
:
return
new
llama_model_ernie4_5
(params);
case
LLM_ARCH_ERNIE4_5_MOE
:
return
new
llama_model_ernie4_5_moe
(params);
case
LLM_ARCH_PADDLEOCR
:
return
new
llama_model_paddleocr
(params);
case
LLM_ARCH_HUNYUAN_MOE
:
return
new
llama_model_hunyuan_moe
(params);
case
LLM_ARCH_HUNYUAN_VL
:
return
new
llama_model_hunyuan_vl
(params);
case
LLM_ARCH_HUNYUAN_DENSE
:
return
new
llama_model_hunyuan_dense
(params);
case
LLM_ARCH_SMOLLM3
:
return
new
llama_model_smollm3
(params);
case
LLM_ARCH_OPENAI_MOE
:
return
new
llama_model_openai_moe
(params);
case
LLM_ARCH_FALCON_H1
:
return
new
llama_model_falcon_h1
(params);
case
LLM_ARCH_LFM2
:
return
new
llama_model_lfm2
(params);
case
LLM_ARCH_LFM2MOE
:
return
new
llama_model_lfm2moe
(params);
case
LLM_ARCH_SMALLTHINKER
:
return
new
llama_model_smallthinker
(params);
case
LLM_ARCH_GROVEMOE
:
return
new
llama_model_grovemoe
(params);
case
LLM_ARCH_APERTUS
:
return
new
llama_model_apertus
(params);
case
LLM_ARCH_MINIMAX_M2
:
return
new
llama_model_minimax_m2
(params);
case
LLM_ARCH_COGVLM
:
return
new
llama_model_cogvlm
(params);
case
LLM_ARCH_PANGU_EMBED
:
return
new
llama_model_pangu_embed
(params);
case
LLM_ARCH_QWEN3NEXT
:
return
new
llama_model_qwen3next
(params);
case
LLM_ARCH_QWEN35
:
return
new
llama_model_qwen35
(params);
case
LLM_ARCH_QWEN35MOE
:
return
new
llama_model_qwen35moe
(params);
case
LLM_ARCH_MISTRAL3
:
return
new
llama_model_mistral3
(params);
case
LLM_ARCH_EAGLE3
:
return
new
llama_model_eagle3
(params);
case
LLM_ARCH_DFLASH
:
return
new
llama_model_dflash
(params);
case
LLM_ARCH_MIMO2
:
return
new
llama_model_mimo2
(params);
case
LLM_ARCH_KIMI_LINEAR
:
return
new
llama_model_kimi_linear
(params);
case
LLM_ARCH_STEP35
:
return
new
llama_model_step35
(params);
default
:
throw
std::runtime_error
(
std::string
(
"
unsupported model architecture: '
"
) +
llm_arch_name
(arch) +
"
'
"
);
}
}
llama_model *
llama_model_create
(llm_arch arch,
const
llama_model_params & params) {
llama_model * model =
llama_model_mapping
(arch, params);
if
(model !=
nullptr
) {
model->
arch
= arch;
auto
& devices = model->
devices
;
if
(!devices.
empty
() && devices[
0
].
is_meta
&& !
llm_arch_supports_sm_tensor
(arch)) {
throw
std::runtime_error
(
std::string
(
"
LLAMA_SPLIT_MODE_TENSOR not implemented for architecture '
"
) +
llm_arch_name
(arch) +
"
'
"
);
}
}
return
model;
}
llama_model *
llama_model_create
(llama_model_loader & ml,
const
llama_model_params & params) {
llm_arch arch = ml.
get_arch
();
if
(arch ==
LLM_ARCH_UNKNOWN
) {
throw
std::runtime_error
(
"
unknown model architecture: '
"
+ ml.
get_arch_name
() +
"
'
"
);
}
return
llama_model_create
(arch, params);
}
struct
ggml_backend_meta_split_state
llama_meta_device_get_split_state
(
const
struct
ggml_tensor
* tensor,
void
* userdata) {
const
llama_meta_device_get_split_state_userdata * ud = (
const
llama_meta_device_get_split_state_userdata *) userdata;
const
llama_hparams & hparams = ud->
model
->
hparams
;
const
std::string tensor_name = tensor->
name
;
const
std::regex
pattern_q_weight
(
"
blk
\\
.
\\
d*
\\
.attn_q.weight
"
);
const
std::regex
pattern_kv_weight
(
"
blk
\\
.
\\
d*
\\
.attn_(k|v).weight
"
);
const
std::regex
pattern_qkv_weight
(
"
blk
\\
.
\\
d*
\\
.attn_qkv.weight
"
);
const
std::regex
pattern_q_bias
(
"
blk
\\
.
\\
d*
\\
.attn_q
\\
.bias
"
);
const
std::regex
pattern_kv_bias
(
"
blk
\\
.
\\
d*
\\
.attn_(k|v)
\\
.bias
"
);
const
std::regex
pattern_qkv_bias
(
"
blk
\\
.
\\
d*
\\
.attn_qkv.bias
"
);
const
std::regex
pattern_qk_norm
(
"
blk
\\
.
\\
d*
\\
.attn_(q|k)_norm
\\
.weight
"
);
const
std::regex
pattern_kv_cache
(
"
cache_(k|v)_l
\\
d*
"
);
const
std::regex
pattern_attn_sinks
(
"
blk
\\
.
\\
d*
\\
.attn_sinks.weight
"
);
const
std::regex
pattern_attn_out_weight
(
"
blk
\\
.
\\
d*
\\
.attn_output.weight
"
);
const
std::regex
pattern_attn_out_bias
(
"
blk
\\
.
\\
d*
\\
.attn_output.bias
"
);
const
std::regex
pattern_attn_gate_weight
(
"
blk
\\
.
\\
d*
\\
.attn_gate.weight
"
);
const
std::regex
pattern_ssm_dt
(
"
blk
\\
.
\\
d*
\\
.ssm_dt.bias
"
);
const
std::regex
pattern_ssm_a
(
"
blk
\\
.
\\
d*
\\
.ssm_a
"
);
const
std::regex
pattern_ssm_alpha
(
"
blk
\\
.
\\
d*
\\
.ssm_alpha.weight
"
);
const
std::regex
pattern_ssm_beta
(
"
blk
\\
.
\\
d*
\\
.ssm_beta.weight
"
);
const
std::regex
pattern_ssm_beta_alpha
(
"
blk
\\
.
\\
d*
\\
.ssm_ba.weight
"
);
const
std::regex
pattern_r_cache
(
"
cache_r_l
\\
d*
"
);
const
std::regex
pattern_s_cache
(
"
cache_s_l
\\
d*
"
);
const
std::regex
pattern_ssm_conv1d
(
"
blk
\\
.
\\
d*
\\
.ssm_conv1d.weight
"
);
const
std::regex
pattern_ssm_out_weight
(
"
blk
\\
.
\\
d*
\\
.ssm_out.weight
"
);
const
std::regex
pattern_ffn_up_gate_weight
(
"
blk
\\
.
\\
d*
\\
.ffn_(up|gate)(_exps)?.weight
"
);
const
std::regex
pattern_ffn_up_gate_bias
(
"
blk
\\
.
\\
d*
\\
.ffn_(up|gate)(_exps)?.bias
"
);
const
std::regex
pattern_ffn_gate_up_weight
(
"
blk
\\
.
\\
d*
\\
.ffn_gate_up(_exps)?.weight
"
);
const
std::regex
pattern_ffn_down_weight
(
"
blk
\\
.
\\
d*
\\
.ffn_down(_exps)?.weight
"
);
const
std::regex
pattern_ffn_down_bias
(
"
blk
\\
.
\\
d*
\\
.ffn_down.bias
"
);
const
std::regex
pattern_ffn_down_exps_bias
(
"
blk
\\
.
\\
d*
\\
.ffn_down_exps.bias
"
);
const
std::regex
pattern_output_weight
(
"
output
\\
.weight
"
);
const
std::regex
pattern_output_bias
(
"
output
\\
.bias
"
);
struct
tensor_config
{
ggml_backend_meta_split_axis axis;
const
ggml_tensor * tensor_axis_0;
uint32_t
il;
size_t
rotation;
//
when assigning tensor slices, rotate how the rounding is done for more even allocation
};
auto
get_tensor_config_impl = [&](
const
ggml_backend_meta_split_axis axis,
const
std::string & suffix =
"
"
,
const
std::string & suffix_fallback =
"
"
) -> tensor_config {
//
the layers in a tensor can be inhomogeneous, if the pattern is cleanly divided by the number of GPUs there can be aliasing effects,
//
count only the same type of previous layers to avoid this
auto
get_il_eff = [&](
const
size_t
il){
size_t
ret =
0
;
const
bool
il_is_recr = hparams.
is_recr
(il);
const
bool
il_is_swa = hparams.
is_swa
(il);
for
(
size_t
il_prev =
0
; il_prev < il; il_prev++) {
ret += hparams.
is_recr
(il_prev) == il_is_recr && hparams.
is_swa
(il_prev) == il_is_swa;
}
return
ret;
};
uint32_t
il;
std::string prefix;
size_t
rotation;
if
(tensor_name.
substr
(
0
,
4
) ==
"
blk.
"
) {
const
size_t
length_prefix = tensor_name.
find
(
'
.
'
,
4
);
GGML_ASSERT
(length_prefix != std::string::npos);
prefix = tensor_name.
substr
(
0
, length_prefix +
1
);
il =
std::stoull
(tensor_name.
substr
(
4
, length_prefix));
rotation =
get_il_eff
(il) % ud->
n_devices
;
}
else
if
(tensor_name.
substr
(
0
,
6
) ==
"
cache_
"
) {
const
size_t
layer_index_start = tensor_name.
find
(
"
_l
"
,
6
);
GGML_ASSERT
(layer_index_start != std::string::npos);
il =
std::stoull
(tensor_name.
substr
(layer_index_start +
2
));
prefix =
"
blk.
"
+
std::to_string
(il) +
"
.
"
;
rotation =
get_il_eff
(il) % ud->
n_devices
;
}
else
{
il =
0
;
rotation = hparams.
n_layer
() % ud->
n_devices
;
}
const
ggml_tensor * tensor_axis_0 = suffix.
empty
() ? tensor : ud->
model
->
get_tensor
((prefix + suffix).
c_str
());
if
(tensor_axis_0 ==
nullptr
) {
GGML_ASSERT
(!suffix_fallback.
empty
());
tensor_axis_0 = ud->
model
->
get_tensor
((prefix + suffix_fallback).
c_str
());
}
GGML_ASSERT
(tensor_axis_0 !=
nullptr
);
return
{axis, tensor_axis_0, il, rotation};
};
auto
get_tensor_config = [&]() -> tensor_config {
//
standard attention
if
(
std::regex_match
(tensor_name, pattern_q_weight) ||
std::regex_match
(tensor_name, pattern_kv_weight)) {
return
get_tensor_config_impl
(
GGML_BACKEND_SPLIT_AXIS_1
,
"
attn_output.weight
"
,
"
ssm_out.weight
"
);
}
if
(
std::regex_match
(tensor_name, pattern_q_bias) ||
std::regex_match
(tensor_name, pattern_kv_bias)) {
return
get_tensor_config_impl
(
GGML_BACKEND_SPLIT_AXIS_0
,
"
attn_output.weight
"
,
"
ssm_out.weight
"
);
}
if
(
std::regex_match
(tensor_name, pattern_qkv_weight)) {
return
get_tensor_config_impl
(
GGML_BACKEND_SPLIT_AXIS_1
,
"
attn_output.weight
"
,
"
ssm_out.weight
"
);
}
if
(
std::regex_match
(tensor_name, pattern_qkv_bias)) {
return
get_tensor_config_impl
(
GGML_BACKEND_SPLIT_AXIS_0
,
"
attn_output.weight
"
,
"
ssm_out.weight
"
);
}
if
(
std::regex_match
(tensor_name, pattern_qk_norm)) {
return
get_tensor_config_impl
(tensor->
ne
[
1
] ==
1
?
GGML_BACKEND_SPLIT_AXIS_MIRRORED
:
GGML_BACKEND_SPLIT_AXIS_1
,
"
attn_output.weight
"
);
}
if
(
std::regex_match
(tensor_name, pattern_kv_cache) ||
std::regex_match
(tensor_name, pattern_attn_sinks)) {
return
get_tensor_config_impl
(
GGML_BACKEND_SPLIT_AXIS_0
,
"
attn_output.weight
"
);
}
if
(
std::regex_match
(tensor_name, pattern_attn_out_weight)) {
return
get_tensor_config_impl
(
GGML_BACKEND_SPLIT_AXIS_0
);
}
if
(
std::regex_match
(tensor_name, pattern_attn_out_bias)) {
return
get_tensor_config_impl
(
GGML_BACKEND_SPLIT_AXIS_MIRRORED
);
}
if
(
std::regex_match
(tensor_name, pattern_attn_gate_weight)) {
return
get_tensor_config_impl
(
GGML_BACKEND_SPLIT_AXIS_1
,
"
attn_output.weight
"
,
"
ssm_out.weight
"
);
}
if
(
std::regex_match
(tensor_name, pattern_ssm_dt) ||
std::regex_match
(tensor_name, pattern_ssm_a)) {
return
get_tensor_config_impl
(
GGML_BACKEND_SPLIT_AXIS_0
,
"
ssm_out.weight
"
);
}
if
(
std::regex_match
(tensor_name, pattern_ssm_alpha) ||
std::regex_match
(tensor_name, pattern_ssm_beta) ||
std::regex_match
(tensor_name, pattern_ssm_beta_alpha)) {
return
get_tensor_config_impl
(
GGML_BACKEND_SPLIT_AXIS_1
,
"
ssm_out.weight
"
);
}
if
(
std::regex_match
(tensor_name, pattern_r_cache) ||
std::regex_match
(tensor_name, pattern_s_cache)) {
return
get_tensor_config_impl
(
GGML_BACKEND_SPLIT_AXIS_0
,
"
ssm_out.weight
"
);
}
if
(
std::regex_match
(tensor_name, pattern_ssm_conv1d)) {
return
get_tensor_config_impl
(
GGML_BACKEND_SPLIT_AXIS_1
,
"
ssm_out.weight
"
);
}
if
(
std::regex_match
(tensor_name, pattern_ssm_out_weight)) {
return
get_tensor_config_impl
(
GGML_BACKEND_SPLIT_AXIS_0
);
}
//
FFN
if
(
std::regex_match
(tensor_name, pattern_ffn_up_gate_weight)) {
return
get_tensor_config_impl
(
GGML_BACKEND_SPLIT_AXIS_1
,
"
ffn_down.weight
"
,
"
ffn_down_exps.weight
"
);
}
if
(
std::regex_match
(tensor_name, pattern_ffn_up_gate_bias)) {
return
get_tensor_config_impl
(
GGML_BACKEND_SPLIT_AXIS_0
,
"
ffn_down.weight
"
,
"
ffn_down_exps.weight
"
);
}
if
(
std::regex_match
(tensor_name, pattern_ffn_gate_up_weight)) {
return
get_tensor_config_impl
(
GGML_BACKEND_SPLIT_AXIS_1
,
"
ffn_down.weight
"
,
"
ffn_down_exps.weight
"
);
}
if
(
std::regex_match
(tensor_name, pattern_ffn_down_weight)) {
return
get_tensor_config_impl
(
GGML_BACKEND_SPLIT_AXIS_0
,
"
ffn_down.weight
"
,
"
ffn_down_exps.weight
"
);
}
if
(
std::regex_match
(tensor_name, pattern_ffn_down_bias)) {
return
get_tensor_config_impl
(
GGML_BACKEND_SPLIT_AXIS_MIRRORED
);
}
if
(
std::regex_match
(tensor_name, pattern_ffn_down_exps_bias)) {
return
get_tensor_config_impl
(
GGML_BACKEND_SPLIT_AXIS_PARTIAL
);
}
//
output
if
(
std::regex_match
(tensor_name, pattern_output_weight)) {
return
get_tensor_config_impl
(
GGML_BACKEND_SPLIT_AXIS_1
);
}
if
(
std::regex_match
(tensor_name, pattern_output_bias)) {
const
ggml_tensor * output_weight = ud->
model
->
get_tensor
(
"
output.weight
"
);
GGML_ASSERT
(output_weight !=
nullptr
);
return
get_tensor_config_impl
(
GGML_BACKEND_SPLIT_AXIS_0
);
}
//
everything else
return
get_tensor_config_impl
(
GGML_BACKEND_SPLIT_AXIS_MIRRORED
);
};
auto
get_split_segments = [&](
int
axis,
uint32_t
il) -> std::vector<std::pair<
int64_t
,
uint32_t
>> {
if
(ud->
model
->
arch
==
LLM_ARCH_QWEN3NEXT
|| ud->
model
->
arch
==
LLM_ARCH_QWEN35
|| ud->
model
->
arch
==
LLM_ARCH_QWEN35MOE
) {
const
int64_t
head_k_dim = hparams.
ssm_d_state
;
const
int64_t
head_v_dim = hparams.
ssm_d_state
;
const
int64_t
n_k_heads = hparams.
ssm_n_group
;
const
int64_t
n_v_heads = hparams.
ssm_dt_rank
;
const
int64_t
key_dim = head_k_dim * n_k_heads;
const
int64_t
value_dim = head_v_dim * n_v_heads;
//
both Qwen 3 Next and Qwen 3.5 support n_v_heads > n_k_heads but the broadcasting pattern is different:
//
- Qwen 3 Next: [k0_v0, k0_v1, k1_v2, k1_v3] (this is the default split pattern)
//
- Qwen 3.5: [k0_v0, k1_v1, k0_v2, k1_v3] (needs segmenting of V on the scale of K to get the correct pattern)
if
(ud->
model
->
arch
==
LLM_ARCH_QWEN3NEXT
) {
if
(
std::regex_match
(tensor_name, pattern_qkv_weight) ||
std::regex_match
(tensor_name, pattern_ssm_conv1d)) {
GGML_ASSERT
(tensor->
ne
[axis] ==
2
*key_dim + value_dim);
return
{{key_dim,
2
}, {value_dim,
1
}};
}
}
else
{
const
int64_t
head_ratio = n_v_heads / n_k_heads;
if
(
std::regex_match
(tensor_name, pattern_qkv_weight) ||
std::regex_match
(tensor_name, pattern_ssm_conv1d)) {
GGML_ASSERT
(tensor->
ne
[axis] ==
2
*key_dim + value_dim);
return
{{key_dim,
2
+ head_ratio}};
}
if
(
std::regex_match
(tensor_name, pattern_attn_gate_weight) ||
std::regex_match
(tensor_name, pattern_ssm_out_weight)) {
return
{{key_dim, head_ratio}};
}
if
(
std::regex_match
(tensor_name, pattern_ssm_dt) ||
std::regex_match
(tensor_name, pattern_ssm_a) ||
std::regex_match
(tensor_name, pattern_ssm_alpha) ||
std::regex_match
(tensor_name, pattern_ssm_beta)) {
return
{{n_k_heads, head_ratio}};
}
if
(
std::regex_match
(tensor_name, pattern_r_cache)) {
return
{{key_dim * (hparams.
ssm_d_conv
-
1
),
2
+ head_ratio}};
}
if
(
std::regex_match
(tensor_name, pattern_s_cache)) {
return
{{n_k_heads * head_v_dim * head_v_dim, head_ratio}};
}
}
//
the FFN is the same for Qwen 3 Next and Qwen 3.5:
if
(
std::regex_match
(tensor_name, pattern_ffn_gate_up_weight)) {
const
int64_t
n_ff_exp = hparams.
n_ff_exp
;
GGML_ASSERT
(tensor->
ne
[axis] ==
2
*n_ff_exp);
return
{{n_ff_exp,
2
}};
}
return
{{tensor->
ne
[axis],
1
}};
}
if
(
std::regex_match
(tensor_name, pattern_qkv_weight) ||
std::regex_match
(tensor_name, pattern_qkv_bias)) {
const
int64_t
n_embd = hparams.
n_embd
;
const
int64_t
n_embd_gqa = hparams.
n_embd_v_gqa
(il);
GGML_ASSERT
(hparams.
n_embd_k_gqa
() == n_embd_gqa);
GGML_ASSERT
(tensor->
ne
[axis] == n_embd +
2
*n_embd_gqa);
return
{{n_embd,
1
}, {n_embd_gqa,
2
}};
}
if
(
std::regex_match
(tensor_name, pattern_ffn_gate_up_weight)) {
const
int64_t
n_ff_exp = hparams.
n_ff_exp
;
GGML_ASSERT
(tensor->
ne
[axis] ==
2
*n_ff_exp);
return
{{n_ff_exp,
2
}};
}
return
{{tensor->
ne
[axis],
1
}};
};
auto
get_split_granularity = [&](
int64_t
blck_size,
uint32_t
il,
const
std::vector<std::pair<
int64_t
,
uint32_t
>> & segments) -> std::vector<
int64_t
> {
//
for better performance it may make sense to round up blck_size to a higher power of 2 so that more efficient kernels can be used
if
(hparams.
is_recr
(il)) {
//
linear attention
const
int64_t
head_dim = hparams.
ssm_d_state
;
const
int64_t
blck_size_perf =
std::lcm
(blck_size,
128
);
const
int64_t
granularity_qkv =
std::lcm
(blck_size_perf, head_dim);
if
(
std::regex_match
(tensor_name, pattern_qkv_weight) ||
std::regex_match
(tensor_name, pattern_attn_gate_weight) ||
std::regex_match
(tensor_name, pattern_ssm_conv1d) ||
std::regex_match
(tensor_name, pattern_ssm_out_weight)) {
return
std::vector<
int64_t
>(segments.
size
(), granularity_qkv);
}
if
(
std::regex_match
(tensor_name, pattern_ssm_dt) ||
std::regex_match
(tensor_name, pattern_ssm_a) ||
std::regex_match
(tensor_name, pattern_ssm_alpha) ||
std::regex_match
(tensor_name, pattern_ssm_beta)) {
return
std::vector<
int64_t
>(segments.
size
(), granularity_qkv / head_dim);
}
if
(
std::regex_match
(tensor_name, pattern_ssm_beta_alpha)) {
return
std::vector<
int64_t
>(segments.
size
(),
2
* (granularity_qkv / head_dim));
}
if
(
std::regex_match
(tensor_name, pattern_r_cache)) {
return
std::vector<
int64_t
>(segments.
size
(), granularity_qkv * (hparams.
ssm_d_conv
-
1
));
}
if
(
std::regex_match
(tensor_name, pattern_s_cache)) {
return
std::vector<
int64_t
>(segments.
size
(), granularity_qkv * head_dim);
}
}
else
{
//
regular attention
const
uint32_t
n_gqa = hparams.
n_gqa
(il);
const
uint32_t
n_embd_q = n_gqa * hparams.
n_embd_head_k
(il);
//
to handle head sizes like 80, only increase granularity while it doesn't cause underutilization
int64_t
blck_size_perf = blck_size;
while
(blck_size_perf <
128
&& blck_size_perf*ud->
n_devices
< n_embd_q) {
blck_size_perf *=
2
;
}
if
(
std::regex_match
(tensor_name, pattern_attn_sinks)) {
GGML_ASSERT
(segments.
size
() ==
1
);
return
{
std::lcm
(n_embd_q, blck_size_perf)/n_embd_q * n_gqa};
}
const
int64_t
granularity_q =
std::lcm
(n_embd_q, blck_size_perf);
if
(
std::regex_match
(tensor_name, pattern_q_weight) ||
std::regex_match
(tensor_name, pattern_q_bias)) {
GGML_ASSERT
(segments.
size
() ==
1
);
//
some models have Q gate tensors, for those cases the granularity needs to be doubled:
if
(ud->
model
->
arch
==
LLM_ARCH_QWEN3NEXT
|| ud->
model
->
arch
==
LLM_ARCH_QWEN35
|| ud->
model
->
arch
==
LLM_ARCH_QWEN35MOE
) {
return
{
std::lcm
(
2
*n_embd_q, blck_size_perf)};
}
return
{granularity_q};
}
if
(
std::regex_match
(tensor_name, pattern_attn_out_weight)) {
GGML_ASSERT
(segments.
size
() ==
1
);
return
{granularity_q};
}
const
int64_t
granularity_kv = granularity_q / n_gqa;
if
(
std::regex_match
(tensor_name, pattern_kv_weight) ||
std::regex_match
(tensor_name, pattern_kv_bias) ||
std::regex_match
(tensor_name, pattern_kv_cache)) {
GGML_ASSERT
(segments.
size
() ==
1
);
return
{granularity_kv};
}
if
(
std::regex_match
(tensor_name, pattern_qkv_weight) ||
std::regex_match
(tensor_name, pattern_qkv_bias)) {
GGML_ASSERT
(segments.
size
() ==
2
);
return
{granularity_q, granularity_kv};
}
}
//
FFN
if
(
std::regex_match
(tensor_name, pattern_ffn_up_gate_weight) ||
std::regex_match
(tensor_name, pattern_ffn_up_gate_bias) ||
std::regex_match
(tensor_name, pattern_ffn_gate_up_weight) ||
std::regex_match
(tensor_name, pattern_ffn_down_weight)) {
const
int64_t
blck_size_perf =
std::lcm
(blck_size,
128
);
GGML_ASSERT
(segments.
size
() ==
1
);
return
{blck_size_perf};
}
//
everything else
GGML_ASSERT
(segments.
size
() ==
1
);
return
{
1
};
};
ggml_backend_meta_split_state split_state;
memset
(&split_state,
0
,
sizeof
(split_state));
tensor_config tc =
get_tensor_config
();
split_state.
axis
= tc.
axis
;
if
(split_state.
axis
>=
0
&& split_state.
axis
<
GGML_MAX_DIMS
) {
const
int64_t
blck_size =
ggml_blck_size
(tc.
tensor_axis_0
->
type
);
const
float
* tensor_split = ud->
model
->
tensor_split
();
std::vector<
float
> tensor_split_scan;
tensor_split_scan.
reserve
(ud->
n_devices
);
for
(
size_t
j =
0
; j < ud->
n_devices
; j++) {
tensor_split_scan.
push_back
(tensor_split ==
nullptr
?
0
.
0f
: tensor_split[(j + tc.
rotation
) % ud->
n_devices
]);
if
(j >
0
) {
tensor_split_scan[j] += tensor_split_scan[j -
1
];
}
}
const
std::vector<std::pair<
int64_t
,
uint32_t
>> segments =
get_split_segments
(split_state.
axis
, tc.
il
);
const
std::vector<
int64_t
> granularity =
get_split_granularity
(blck_size, tc.
il
, segments);
for
(
size_t
is =
0
; is < segments.
size
(); is++) {
const
int64_t
ne_s = segments[is].
first
;
const
uint32_t
nr_s = segments[is].
second
;
const
int64_t
g_s = granularity[is];
int64_t
low =
0
;
size_t
j =
0
;
for
(; j < ud->
n_devices
-
1
; j++) {
int64_t
high = tensor_split_scan.
back
() ==
0
.
0f
?
ne_s * (j+
1
)/ud->
n_devices
: ne_s * tensor_split_scan[j]/tensor_split_scan.
back
();
if
(high % g_s !=
0
) {
high -= high % g_s;
}
split_state.
ne
[is*ud->
n_devices
+ (j + tc.
rotation
) % ud->
n_devices
] = high - low;
low = high;
}
split_state.
ne
[is*ud->
n_devices
+ (j + tc.
rotation
) % ud->
n_devices
] = ne_s - low;
split_state.
nr
[is] = nr_s;
}
split_state.
n_segments
= segments.
size
();
}
else
{
memset
(split_state.
ne
,
0
,
sizeof
(split_state.
ne
));
split_state.
nr
[
0
] =
1
;
split_state.
n_segments
=
1
;
}
return
split_state;
GGML_UNUSED
(userdata);
}
const
char
*
llm_type_name
(llm_type type) {
switch
(type) {
case
LLM_TYPE_14M
:
return
"
14M
"
;
case
LLM_TYPE_17M
:
return
"
17M
"
;
case
LLM_TYPE_22M
:
return
"
22M
"
;
case
LLM_TYPE_33M
:
return
"
33M
"
;
case
LLM_TYPE_47M
:
return
"
47M
"
;
case
LLM_TYPE_60M
:
return
"
60M
"
;
case
LLM_TYPE_70M
:
return
"
70M
"
;
case
LLM_TYPE_80M
:
return
"
80M
"
;
case
LLM_TYPE_109M
:
return
"
109M
"
;
case
LLM_TYPE_137M
:
return
"
137M
"
;
case
LLM_TYPE_140M
:
return
"
140M
"
;
case
LLM_TYPE_149M
:
return
"
149M
"
;
case
LLM_TYPE_160M
:
return
"
160M
"
;
case
LLM_TYPE_190M
:
return
"
190M
"
;
case
LLM_TYPE_220M
:
return
"
220M
"
;
case
LLM_TYPE_250M
:
return
"
250M
"
;
case
LLM_TYPE_256M
:
return
"
256M
"
;
case
LLM_TYPE_270M
:
return
"
270M
"
;
case
LLM_TYPE_335M
:
return
"
335M
"
;
case
LLM_TYPE_350M
:
return
"
350M
"
;
case
LLM_TYPE_360M
:
return
"
360M
"
;
case
LLM_TYPE_395M
:
return
"
395M
"
;
case
LLM_TYPE_410M
:
return
"
410M
"
;
case
LLM_TYPE_450M
:
return
"
450M
"
;
case
LLM_TYPE_475M
:
return
"
475M
"
;
case
LLM_TYPE_558M
:
return
"
558M
"
;
case
LLM_TYPE_700M
:
return
"
700M
"
;
case
LLM_TYPE_770M
:
return
"
770M
"
;
case
LLM_TYPE_780M
:
return
"
780M
"
;
case
LLM_TYPE_950M
:
return
"
950M
"
;
case
LLM_TYPE_0_3B
:
return
"
0.3B
"
;
case
LLM_TYPE_0_5B
:
return
"
0.5B
"
;
case
LLM_TYPE_0_6B
:
return
"
0.6B
"
;
case
LLM_TYPE_0_8B
:
return
"
0.8B
"
;
case
LLM_TYPE_1B
:
return
"
1B
"
;
case
LLM_TYPE_1_2B
:
return
"
1.2B
"
;
case
LLM_TYPE_1_3B
:
return
"
1.3B
"
;
case
LLM_TYPE_1_4B
:
return
"
1.4B
"
;
case
LLM_TYPE_1_5B
:
return
"
1.5B
"
;
case
LLM_TYPE_1_6B
:
return
"
1.6B
"
;
case
LLM_TYPE_1_7B
:
return
"
1.7B
"
;
case
LLM_TYPE_1_8B
:
return
"
1.8B
"
;
case
LLM_TYPE_2B
:
return
"
2B
"
;
case
LLM_TYPE_2_6B
:
return
"
2.6B
"
;
case
LLM_TYPE_2_8B
:
return
"
2.8B
"
;
case
LLM_TYPE_2_9B
:
return
"
2.9B
"
;
case
LLM_TYPE_3B
:
return
"
3B
"
;
case
LLM_TYPE_4B
:
return
"
4B
"
;
case
LLM_TYPE_6B
:
return
"
6B
"
;
case
LLM_TYPE_6_9B
:
return
"
6.9B
"
;
case
LLM_TYPE_7B
:
return
"
7B
"
;
case
LLM_TYPE_8B
:
return
"
8B
"
;
case
LLM_TYPE_9B
:
return
"
9B
"
;
case
LLM_TYPE_11B
:
return
"
11B
"
;
case
LLM_TYPE_12B
:
return
"
12B
"
;
case
LLM_TYPE_13B
:
return
"
13B
"
;
case
LLM_TYPE_14B
:
return
"
14B
"
;
case
LLM_TYPE_15B
:
return
"
15B
"
;
case
LLM_TYPE_16B
:
return
"
16B
"
;
case
LLM_TYPE_20B
:
return
"
20B
"
;
case
LLM_TYPE_26B
:
return
"
26B
"
;
case
LLM_TYPE_27B
:
return
"
27B
"
;
case
LLM_TYPE_30B
:
return
"
30B
"
;
case
LLM_TYPE_31B
:
return
"
31B
"
;
case
LLM_TYPE_32B
:
return
"
32B
"
;
case
LLM_TYPE_34B
:
return
"
34B
"
;
case
LLM_TYPE_35B
:
return
"
35B
"
;
case
LLM_TYPE_36B
:
return
"
36B
"
;
case
LLM_TYPE_40B
:
return
"
40B
"
;
case
LLM_TYPE_65B
:
return
"
65B
"
;
case
LLM_TYPE_70B
:
return
"
70B
"
;
case
LLM_TYPE_120B
:
return
"
120B
"
;
case
LLM_TYPE_142B
:
return
"
142B
"
;
case
LLM_TYPE_236B
:
return
"
236B
"
;
case
LLM_TYPE_290B
:
return
"
290B
"
;
case
LLM_TYPE_314B
:
return
"
314B
"
;
case
LLM_TYPE_405B
:
return
"
405B
"
;
case
LLM_TYPE_671B
:
return
"
671B
"
;
case
LLM_TYPE_SMALL
:
return
"
0.1B
"
;
case
LLM_TYPE_MEDIUM
:
return
"
0.4B
"
;
case
LLM_TYPE_LARGE
:
return
"
0.8B
"
;
case
LLM_TYPE_XL
:
return
"
1.5B
"
;
case
LLM_TYPE_A1_7B
:
return
"
A1.7B
"
;
case
LLM_TYPE_A2_7B
:
return
"
A2.7B
"
;
case
LLM_TYPE_8x7B:
return
"
8x7B
"
;
case
LLM_TYPE_8x22B:
return
"
8x22B
"
;
case
LLM_TYPE_16x12B:
return
"
16x12B
"
;
case
LLM_TYPE_16x3_8B:
return
"
16x3.8B
"
;
case
LLM_TYPE_10B_128x3_66B:
return
"
10B+128x3.66B
"
;
case
LLM_TYPE_57B_A14B
:
return
"
57B.A14B
"
;
case
LLM_TYPE_17B_16E
:
return
"
17Bx16E (Scout)
"
;
case
LLM_TYPE_17B_128E
:
return
"
17Bx128E (Maverick)
"
;
case
LLM_TYPE_A13B
:
return
"
A13B
"
;
case
LLM_TYPE_7B_A1B
:
return
"
7B.A1B
"
;
case
LLM_TYPE_8B_A1B
:
return
"
8B.A1B
"
;
case
LLM_TYPE_12B_A2_5B
:
return
"
12B.A2.5B
"
;
case
LLM_TYPE_16B_A1B
:
return
"
16B.A1B
"
;
case
LLM_TYPE_21B_A3B
:
return
"
21B.A3B
"
;
case
LLM_TYPE_24B_A2B
:
return
"
24B.A2B
"
;
case
LLM_TYPE_26B_A4B
:
return
"
26B.A4B
"
;
case
LLM_TYPE_30B_A3B
:
return
"
30B.A3B
"
;
case
LLM_TYPE_31B_A3_5B
:
return
"
31B.A3.5B
"
;
case
LLM_TYPE_35B_A3B
:
return
"
35B.A3B
"
;
case
LLM_TYPE_48B_A3B
:
return
"
48B.A3B
"
;
case
LLM_TYPE_80B_A3B
:
return
"
80B.A3B
"
;
case
LLM_TYPE_100B_A6B
:
return
"
100B.A6B
"
;
case
LLM_TYPE_102B_A12B
:
return
"
102B.A12B
"
;
case
LLM_TYPE_106B_A12B
:
return
"
106B.A12B
"
;
case
LLM_TYPE_120B_A12B
:
return
"
120B.A12B
"
;
case
LLM_TYPE_122B_A10B
:
return
"
122B.A10B
"
;
case
LLM_TYPE_196B_A11B
:
return
"
196B.A11B
"
;
case
LLM_TYPE_230B_A10B
:
return
"
230B.A10B
"
;
case
LLM_TYPE_235B_A22B
:
return
"
235B.A22B
"
;
case
LLM_TYPE_300B_A47B
:
return
"
300B.A47B
"
;
case
LLM_TYPE_310B_A15B
:
return
"
310B.A15B
"
;
case
LLM_TYPE_355B_A32B
:
return
"
355B.A32B
"
;
case
LLM_TYPE_397B_A17B
:
return
"
397B.A17B
"
;
case
LLM_TYPE_685B_A37B
:
return
"
685B.A37B
"
;
case
LLM_TYPE_744B_A40B
:
return
"
744B.A40B
"
;
case
LLM_TYPE_E2B
:
return
"
E2B
"
;
case
LLM_TYPE_E4B
:
return
"
E4B
"
;
default
:
return
"
?B
"
;
}
}
static
const
char
*
llama_expert_gating_func_name
(llama_expert_gating_func_type type) {
switch
(type) {
case
LLAMA_EXPERT_GATING_FUNC_TYPE_SOFTMAX
:
return
"
softmax
"
;
case
LLAMA_EXPERT_GATING_FUNC_TYPE_SIGMOID
:
return
"
sigmoid
"
;
default
:
return
"
unknown
"
;
}
}
static
const
std::map<llama_rope_scaling_type,
const
char
*>
LLAMA_ROPE_SCALING_TYPES
= {
{
LLAMA_ROPE_SCALING_TYPE_NONE
,
"
none
"
},
{
LLAMA_ROPE_SCALING_TYPE_LINEAR
,
"
linear
"
},
{
LLAMA_ROPE_SCALING_TYPE_YARN
,
"
yarn
"
},
{
LLAMA_ROPE_SCALING_TYPE_LONGROPE
,
"
longrope
"
},
};
std::string
llama_rope_scaling_type_name
(llama_rope_scaling_type rope_scaling_type) {
return
LLAMA_ROPE_SCALING_TYPES
.
at
(rope_scaling_type);
}
static
llama_rope_scaling_type
llama_rope_scaling_type_from_string
(
const
std::string & name) {
for
(
const
auto
& kv :
LLAMA_ROPE_SCALING_TYPES
) {
if
(kv.
second
== name) {
return
(llama_rope_scaling_type) kv.
first
;
}
}
return
LLAMA_ROPE_SCALING_TYPE_UNSPECIFIED
;
}
//
Maps the GGUF `<arch>.hidden_activation` string to the FFN op type used by the
//
graph builders. Only gated activations that map cleanly to llm_ffn_op_type are
//
listed; unrecognized values fall back to GeGLU, which matches the historical
//
default for ModernBert-style architectures.
static
const
std::map<std::string, llm_ffn_op_type>
LLM_FFN_OP_TYPES_FROM_STRING
= {
{
"
gelu
"
,
LLM_FFN_GEGLU
},
{
"
geglu
"
,
LLM_FFN_GEGLU
},
{
"
silu
"
,
LLM_FFN_SWIGLU
},
{
"
swish
"
,
LLM_FFN_SWIGLU
},
{
"
swiglu
"
,
LLM_FFN_SWIGLU
},
{
"
relu
"
,
LLM_FFN_RELU
},
{
"
reglu
"
,
LLM_FFN_REGLU
},
};
llm_ffn_op_type
llm_ffn_op_type_from_string
(
const
std::string & name, llm_ffn_op_type fallback) {
const
auto
it =
LLM_FFN_OP_TYPES_FROM_STRING
.
find
(name);
if
(it !=
LLM_FFN_OP_TYPES_FROM_STRING
.
end
()) {
return
it->
second
;
}
return
fallback;
}
//
CPU: ACCEL -> GPU host -> CPU extra -> CPU
static
buft_list_t
make_cpu_buft_list
(
const
std::vector<llama_device> & devices,
bool
use_extra_bufts,
bool
no_host) {
buft_list_t
buft_list;
//
add ACCEL buffer types
for
(
size_t
i =
0
; i <
ggml_backend_dev_count
(); ++i) {
ggml_backend_dev_t
dev =
ggml_backend_dev_get
(i);
if
(
ggml_backend_dev_type
(dev) ==
GGML_BACKEND_DEVICE_TYPE_ACCEL
) {
auto
* buft =
ggml_backend_dev_buffer_type
(dev);
//
skip
if
(buft !=
ggml_backend_cpu_buffer_type
()) {
buft_list.
emplace_back
(dev, buft);
}
}
}
//
add a host buffer type
//
storing the tensors in a host buffer is useful when the processing of large batches
//
is offloaded to a GPU device, since it reduces the time spent on data transfers
//
generally, this will be done using the first device in the list
//
a better approach would be to handle this on a weight-by-weight basis using the offload_op
//
function of the device to determine if it would benefit from being stored in a host buffer
if
(!no_host) {
for
(
const
auto
& dev : devices) {
ggml_backend_buffer_type_t
buft =
ggml_backend_dev_host_buffer_type
(dev.
dev
);
if
(buft) {
buft_list.
emplace_back
(dev.
dev
, buft);
break
;
}
}
}
//
add extra buffer types
if
(use_extra_bufts) {
auto
* cpu_dev =
ggml_backend_dev_by_type
(
GGML_BACKEND_DEVICE_TYPE_CPU
);
if
(cpu_dev ==
nullptr
) {
throw
std::runtime_error
(
format
(
"
%s: no CPU backend found
"
, __func__));
}
auto
* cpu_reg =
ggml_backend_dev_backend_reg
(cpu_dev);
auto
ggml_backend_dev_get_extra_bufts_fn = (
ggml_backend_dev_get_extra_bufts_t
)
ggml_backend_reg_get_proc_address
(cpu_reg,
"
ggml_backend_dev_get_extra_bufts
"
);
if
(ggml_backend_dev_get_extra_bufts_fn) {
ggml_backend_buffer_type_t
* extra_bufts =
ggml_backend_dev_get_extra_bufts_fn
(cpu_dev);
while
(extra_bufts && *extra_bufts) {
buft_list.
emplace_back
(cpu_dev, *extra_bufts);
++extra_bufts;
}
}
}
//
add the CPU buffer type
for
(
size_t
i =
0
; i <
ggml_backend_dev_count
(); ++i) {
ggml_backend_dev_t
dev =
ggml_backend_dev_get
(i);
if
(
ggml_backend_dev_type
(dev) ==
GGML_BACKEND_DEVICE_TYPE_CPU
) {
buft_list.
emplace_back
(dev,
ggml_backend_dev_buffer_type
(dev));
}
}
return
buft_list;
}
//
GPU: split if LLAMA_SPLIT_MODE_ROW -> GPU
static
buft_list_t
make_gpu_buft_list
(
ggml_backend_dev_t
dev, llama_split_mode split_mode,
const
float
* tensor_split) {
buft_list_t
buft_list;
//
add the device split buffer type if requested and available
if
(split_mode ==
LLAMA_SPLIT_MODE_ROW
) {
ggml_backend_reg_t
reg =
ggml_backend_dev_backend_reg
(dev);
auto
ggml_backend_split_buffer_type_fn = (
ggml_backend_split_buffer_type_t
)
ggml_backend_reg_get_proc_address
(reg,
"
ggml_backend_split_buffer_type
"
);
if
(ggml_backend_split_buffer_type_fn) {
size_t
dev_index = [&]() {
auto
* reg =
ggml_backend_dev_backend_reg
(dev);
for
(
size_t
i =
0
; i <
ggml_backend_reg_dev_count
(reg); ++i) {
if
(
ggml_backend_reg_dev_get
(reg, i) == dev) {
return
i;
}
}
throw
std::runtime_error
(
format
(
"
device %s not found in its backend reg
"
,
ggml_backend_dev_name
(dev)));
}();
auto
* buft =
ggml_backend_split_buffer_type_fn
(dev_index, tensor_split);
if
(buft !=
nullptr
) {
buft_list.
emplace_back
(dev, buft);
}
}
}
//
add the device default buffer type
buft_list.
emplace_back
(dev,
ggml_backend_dev_buffer_type
(dev));
//
add the device extra buffer type (if any)
ggml_backend_reg_t
reg =
ggml_backend_dev_backend_reg
(dev);
if
(reg) {
auto
ggml_backend_dev_get_extra_bufts_fn = (
ggml_backend_dev_get_extra_bufts_t
)
ggml_backend_reg_get_proc_address
(reg,
"
ggml_backend_dev_get_extra_bufts
"
);
if
(ggml_backend_dev_get_extra_bufts_fn) {
ggml_backend_buffer_type_t
* extra_bufts =
ggml_backend_dev_get_extra_bufts_fn
(dev);
while
(extra_bufts && *extra_bufts) {
buft_list.
emplace_back
(dev, *extra_bufts);
++extra_bufts;
}
}
}
return
buft_list;
}
struct
llama_model
::impl {
impl
() =
default
;
~impl
() =
default
;
uint64_t
n_elements =
0
;
size_t
n_bytes =
0
;
std::string desc_str;
//
model memory mapped files
llama_mmaps mappings;
//
objects representing data potentially being locked in memory
llama_mlocks mlock_bufs;
llama_mlocks mlock_mmaps;
//
contexts where the model tensors metadata is stored as well as the corresponding buffers:
std::vector<std::pair<ggml_context_ptr, std::vector<ggml_backend_buffer_ptr>>> ctxs_bufs;
buft_list_t
cpu_buft_list;
std::map<
ggml_backend_dev_t
,
buft_list_t
> gpu_buft_list;
struct
layer_dev
{
ggml_backend_dev_t
dev;
buft_list_t
* buft_list;
View remainder of file in raw view
Back
|
FazBrowse Home
|
New Git URL