File: ops.md

package info (click to toggle)
llama.cpp 5882%2Bdfsg-3
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 34,020 kB
  • sloc: cpp: 189,548; ansic: 115,889; python: 24,977; objc: 6,050; lisp: 5,741; sh: 5,571; makefile: 1,293; javascript: 807; xml: 259
file content (95 lines) | stat: -rw-r--r-- 5,472 bytes parent folder | download
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
# GGML Operations

List of GGML operations and backend support status.

Legend:
- ✅ Fully supported by this backend
- 🟡 Partially supported by this backend
- ❌ Not supported by this backend

| Operation | BLAS | CPU | CUDA | Metal |
|-----------|------|------|------|------|
|                              ABS | ❌ | ✅ | 🟡 | ❌ |
|                              ACC | ❌ | ✅ | ✅ | ✅ |
|                              ADD | ❌ | ✅ | ✅ | 🟡 |
|                             ADD1 | ❌ | ✅ | ✅ | ❌ |
|                           ARANGE | ❌ | ✅ | ✅ | ✅ |
|                           ARGMAX | ❌ | ✅ | ✅ | ✅ |
|                          ARGSORT | ❌ | ✅ | ✅ | ✅ |
|                            CLAMP | ❌ | ✅ | ✅ | 🟡 |
|                           CONCAT | ❌ | ✅ | 🟡 | ✅ |
|                             CONT | ❌ | ✅ | 🟡 | ✅ |
|                       CONV_2D_DW | ❌ | ✅ | ✅ | ❌ |
|                CONV_TRANSPOSE_1D | ❌ | ✅ | ✅ | ✅ |
|                CONV_TRANSPOSE_2D | ❌ | ✅ | ✅ | ❌ |
|                              COS | ❌ | ✅ | ✅ | 🟡 |
|                      COUNT_EQUAL | ❌ | ✅ | ✅ | ❌ |
|                              CPY | ❌ | 🟡 | 🟡 | 🟡 |
|               CROSS_ENTROPY_LOSS | ❌ | ✅ | ✅ | ❌ |
|          CROSS_ENTROPY_LOSS_BACK | ❌ | ✅ | ✅ | ❌ |
|                    DIAG_MASK_INF | ❌ | ✅ | ✅ | 🟡 |
|                              DIV | ❌ | ✅ | ✅ | 🟡 |
|                              DUP | ❌ | ✅ | 🟡 | 🟡 |
|                              ELU | ❌ | ✅ | ❌ | 🟡 |
|                              EXP | ❌ | ✅ | 🟡 | ❌ |
|                   FLASH_ATTN_EXT | ❌ | ✅ | 🟡 | 🟡 |
|                GATED_LINEAR_ATTN | ❌ | ✅ | ✅ | ❌ |
|                            GEGLU | ❌ | ✅ | ✅ | 🟡 |
|                        GEGLU_ERF | ❌ | ✅ | ✅ | 🟡 |
|                      GEGLU_QUICK | ❌ | ✅ | ✅ | 🟡 |
|                             GELU | ❌ | ✅ | 🟡 | 🟡 |
|                         GELU_ERF | ❌ | ✅ | 🟡 | 🟡 |
|                       GELU_QUICK | ❌ | ✅ | 🟡 | 🟡 |
|                         GET_ROWS | ❌ | ✅ | 🟡 | ✅ |
|                    GET_ROWS_BACK | ❌ | 🟡 | 🟡 | ❌ |
|                       GROUP_NORM | ❌ | ✅ | ✅ | ✅ |
|                      HARDSIGMOID | ❌ | ✅ | 🟡 | ❌ |
|                        HARDSWISH | ❌ | ✅ | 🟡 | ❌ |
|                           IM2COL | ❌ | ✅ | ✅ | 🟡 |
|                          L2_NORM | ❌ | ✅ | ✅ | ✅ |
|                       LEAKY_RELU | ❌ | ✅ | ✅ | ✅ |
|                              LOG | ❌ | ✅ | ✅ | ❌ |
|                             MEAN | ❌ | ✅ | ✅ | ✅ |
|                              MUL | ❌ | ✅ | ✅ | 🟡 |
|                          MUL_MAT | 🟡 | 🟡 | 🟡 | 🟡 |
|                       MUL_MAT_ID | ❌ | ✅ | ✅ | ✅ |
|                              NEG | ❌ | ✅ | 🟡 | 🟡 |
|                             NORM | ❌ | ✅ | ✅ | 🟡 |
|                   OPT_STEP_ADAMW | ❌ | ✅ | ✅ | ❌ |
|                         OUT_PROD | 🟡 | 🟡 | 🟡 | ❌ |
|                              PAD | ❌ | ✅ | ✅ | ✅ |
|                   PAD_REFLECT_1D | ❌ | ✅ | ❌ | ✅ |
|                          POOL_2D | ❌ | ✅ | ✅ | ✅ |
|                            REGLU | ❌ | ✅ | ✅ | 🟡 |
|                             RELU | ❌ | ✅ | 🟡 | 🟡 |
|                           REPEAT | ❌ | ✅ | 🟡 | ✅ |
|                      REPEAT_BACK | ❌ | ✅ | ✅ | ❌ |
|                         RMS_NORM | ❌ | ✅ | ✅ | 🟡 |
|                    RMS_NORM_BACK | ❌ | ✅ | ✅ | ❌ |
|                     RMS_NORM_MUL | ❌ | ✅ | ✅ | ✅ |
|                             ROPE | ❌ | ✅ | ✅ | ✅ |
|                        ROPE_BACK | ❌ | ✅ | ✅ | ❌ |
|                        RWKV_WKV6 | ❌ | ✅ | ✅ | ✅ |
|                        RWKV_WKV7 | ❌ | ✅ | ✅ | ✅ |
|                            SCALE | ❌ | ✅ | ✅ | ✅ |
|                              SET | ❌ | ✅ | ❌ | ✅ |
|                         SET_ROWS | ❌ | 🟡 | ❌ | 🟡 |
|                              SGN | ❌ | ✅ | 🟡 | ❌ |
|                          SIGMOID | ❌ | ✅ | 🟡 | 🟡 |
|                             SILU | ❌ | ✅ | 🟡 | 🟡 |
|                        SILU_BACK | ❌ | ✅ | ✅ | ❌ |
|                              SIN | ❌ | ✅ | ✅ | 🟡 |
|                         SOFT_MAX | ❌ | ✅ | ✅ | ✅ |
|                    SOFT_MAX_BACK | ❌ | 🟡 | 🟡 | ❌ |
|                              SQR | ❌ | ✅ | ✅ | 🟡 |
|                             SQRT | ❌ | ✅ | ✅ | 🟡 |
|                         SSM_CONV | ❌ | ✅ | ✅ | ✅ |
|                         SSM_SCAN | ❌ | ✅ | ✅ | ✅ |
|                             STEP | ❌ | ✅ | 🟡 | ❌ |
|                              SUB | ❌ | ✅ | ✅ | 🟡 |
|                              SUM | ❌ | ✅ | ✅ | ❌ |
|                         SUM_ROWS | ❌ | ✅ | ✅ | ✅ |
|                           SWIGLU | ❌ | ✅ | ✅ | 🟡 |
|                             TANH | ❌ | ✅ | 🟡 | 🟡 |
|               TIMESTEP_EMBEDDING | ❌ | ✅ | ✅ | ✅ |
|                          UPSCALE | ❌ | ✅ | ✅ | 🟡 |