ops.md 7.9 KB

GGML Operations

List of GGML operations and backend support status.

How to add a backend to this table:

  1. Run test-backend-ops support --output csv with your backend name and redirect output to a csv file in docs/ops/ (e.g., docs/ops/CUDA.csv)
  2. Regenerate /docs/ops.md via ./scripts/create_ops_docs.py

Legend:

  • โœ… Fully supported by this backend
  • ๐ŸŸก Partially supported by this backend
  • โŒ Not supported by this backend
Operation BLAS CANN CPU CUDA Metal OpenCL SYCL Vulkan
ABS โŒ โœ… โœ… ๐ŸŸก ๐ŸŸก โŒ ๐ŸŸก โŒ
ACC โŒ โœ… โœ… โœ… โœ… โŒ โœ… โœ…
ADD โŒ โœ… โœ… โœ… ๐ŸŸก ๐ŸŸก โœ… โœ…
ADD1 โŒ โœ… โœ… โœ… โŒ โŒ โœ… โŒ
ARANGE โŒ โœ… โœ… โœ… โœ… โŒ โŒ โŒ
ARGMAX โŒ โœ… โœ… โœ… โœ… โŒ โœ… โœ…
ARGSORT โŒ โœ… โœ… โœ… โœ… โœ… โœ… โœ…
CLAMP โŒ โœ… โœ… โœ… ๐ŸŸก ๐ŸŸก โœ… ๐ŸŸก
CONCAT โŒ โœ… โœ… ๐ŸŸก โœ… ๐ŸŸก ๐ŸŸก โœ…
CONT โŒ ๐ŸŸก โœ… โœ… โœ… ๐ŸŸก ๐ŸŸก ๐ŸŸก
CONV_2D โŒ โŒ โœ… โŒ โŒ โœ… โŒ โœ…
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_ADD โŒ โœ… โœ… โœ… โœ… โœ… โœ… โœ…
ROLL โŒ โŒ โœ… โŒ โŒ โŒ โŒ โœ…
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 โŒ ๐ŸŸก โœ… โœ… ๐ŸŸก โœ… ๐ŸŸก โœ