Skip to content

Commit c35f9ea

Browse files
authored
ops : update Metal (#14912)
1 parent 1f45f28 commit c35f9ea

File tree

2 files changed

+16273
-6542
lines changed

2 files changed

+16273
-6542
lines changed

docs/ops.md

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ Legend:
1414

1515
| Operation | BLAS | CPU | CUDA | Metal | SYCL | Vulkan |
1616
|-----------|------|------|------|------|------|------|
17-
| ABS ||| 🟡 | | 🟡 ||
17+
| ABS ||| 🟡 | 🟡 | 🟡 ||
1818
| ACC |||||||
1919
| ADD |||| 🟡 |||
2020
| ADD1 |||||||
@@ -37,7 +37,7 @@ Legend:
3737
| DIV |||| 🟡 |||
3838
| DUP ||| 🟡 | 🟡 || 🟡 |
3939
| ELU ||| 🟡 | 🟡 | 🟡 ||
40-
| EXP ||| 🟡 | | 🟡 ||
40+
| EXP ||| 🟡 | 🟡 | 🟡 ||
4141
| FLASH_ATTN_EXT ||| 🟡 | 🟡 || 🟡 |
4242
| GATED_LINEAR_ATTN |||||||
4343
| GEGLU |||| 🟡 || 🟡 |
@@ -49,8 +49,8 @@ Legend:
4949
| GET_ROWS ||| 🟡 || 🟡 | 🟡 |
5050
| GET_ROWS_BACK || 🟡 | 🟡 ||||
5151
| GROUP_NORM |||||||
52-
| HARDSIGMOID ||| 🟡 | | 🟡 ||
53-
| HARDSWISH ||| 🟡 | | 🟡 ||
52+
| HARDSIGMOID ||| 🟡 | 🟡 | 🟡 ||
53+
| HARDSWISH ||| 🟡 | 🟡 | 🟡 ||
5454
| IM2COL |||| 🟡 |||
5555
| L2_NORM |||||||
5656
| LEAKY_RELU |||||||
@@ -72,8 +72,8 @@ Legend:
7272
| REPEAT_BACK |||||||
7373
| RMS_NORM |||| 🟡 |||
7474
| RMS_NORM_BACK |||||||
75-
| RMS_NORM_MUL |||| |||
76-
| RMS_NORM_MUL_ADD |||| |||
75+
| RMS_NORM_MUL |||| |||
76+
| RMS_NORM_MUL_ADD |||| |||
7777
| ROLL |||||||
7878
| ROPE |||||||
7979
| ROPE_BACK |||||||
@@ -82,7 +82,7 @@ Legend:
8282
| SCALE |||||||
8383
| SET |||||||
8484
| SET_ROWS || 🟡 | 🟡 | 🟡 | 🟡 | 🟡 |
85-
| SGN ||| 🟡 | | 🟡 ||
85+
| SGN ||| 🟡 | 🟡 | 🟡 ||
8686
| SIGMOID ||| 🟡 | 🟡 | 🟡 | 🟡 |
8787
| SILU ||| 🟡 | 🟡 | 🟡 | 🟡 |
8888
| SILU_BACK |||||||
@@ -93,7 +93,7 @@ Legend:
9393
| SQRT |||| 🟡 |||
9494
| SSM_CONV |||||||
9595
| SSM_SCAN |||||||
96-
| STEP ||| 🟡 | | 🟡 ||
96+
| STEP ||| 🟡 | 🟡 | 🟡 ||
9797
| SUB |||| 🟡 |||
9898
| SUM |||||||
9999
| SUM_ROWS |||||||

0 commit comments

Comments
 (0)