Skip to content

Commit ae2d348

Browse files
authored
metal: Implement ROLL op (ggml-org#21946)
* nix: support unified apple-sdk * Impl roll op for Metal * Revert "nix: support unified apple-sdk" This reverts commit abfa473. * update ops.md * update op docs
1 parent 1e796eb commit ae2d348

File tree

9 files changed

+3072
-856
lines changed

9 files changed

+3072
-856
lines changed

docs/ops.md

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -22,13 +22,13 @@ Legend:
2222
| ARANGE ||||||||||||
2323
| ARGMAX ||||||||||||
2424
| ARGSORT |||||| 🟡 | 🟡 |||||
25-
| CEIL |||| 🟡 | ||| 🟡 ||||
25+
| CEIL |||| 🟡 | ||| 🟡 ||||
2626
| CLAMP |||||| 🟡 | 🟡 | 🟡 ||||
2727
| CONCAT |||| 🟡 || 🟡 ||||||
28-
| CONT || 🟡 ||| 🟡 | 🟡 | 🟡 || 🟡 |||
28+
| CONT || 🟡 ||| | 🟡 | 🟡 || 🟡 |||
2929
| CONV_2D ||||||||||||
3030
| CONV_2D_DW ||||||||||||
31-
| CONV_3D ||||| |||||||
31+
| CONV_3D ||||| |||||||
3232
| CONV_TRANSPOSE_1D ||||||||||||
3333
| CONV_TRANSPOSE_2D ||||||||||||
3434
| COS ||||||| 🟡 | 🟡 ||||
@@ -46,7 +46,7 @@ Legend:
4646
| EXPM1 |||| 🟡 ||||||||
4747
| FILL ||||||||||||
4848
| FLASH_ATTN_EXT || 🟡 || 🟡 | 🟡 | 🟡 | 🟡 | 🟡 | 🟡 |||
49-
| FLOOR |||| 🟡 | || 🟡 | 🟡 ||||
49+
| FLOOR |||| 🟡 | || 🟡 | 🟡 ||||
5050
| GATED_DELTA_NET ||||| 🟡 |||||||
5151
| GATED_LINEAR_ATTN ||||||||||||
5252
| GEGLU ||||| 🟡 ||| 🟡 ||||
@@ -84,10 +84,10 @@ Legend:
8484
| REPEAT_BACK ||||||||||||
8585
| RMS_NORM ||||||||||||
8686
| RMS_NORM_BACK ||||||||||||
87-
| ROLL ||||| |||||||
87+
| ROLL ||||| |||||||
8888
| ROPE ||||||||||||
8989
| ROPE_BACK ||||||||||||
90-
| ROUND |||| 🟡 | || 🟡 | 🟡 ||||
90+
| ROUND |||| 🟡 | || 🟡 | 🟡 ||||
9191
| RWKV_WKV6 ||||||||||||
9292
| RWKV_WKV7 ||||||||||||
9393
| SCALE || 🟡 ||||||||||
@@ -116,6 +116,6 @@ Legend:
116116
| TIMESTEP_EMBEDDING ||||||||||||
117117
| TOP_K ||||||| 🟡 | 🟡 ||||
118118
| TRI ||||||||||||
119-
| TRUNC |||| 🟡 | || 🟡 | 🟡 ||||
119+
| TRUNC |||| 🟡 | || 🟡 | 🟡 ||||
120120
| UPSCALE || 🟡 |||| 🟡 ||||||
121-
| XIELU ||||| |||||||
121+
| XIELU ||||| |||||||

0 commit comments

Comments
 (0)