ops : update Metal (#14912)
This commit is contained in:
parent
1f45f2890e
commit
c35f9eaf09
16
docs/ops.md
16
docs/ops.md
|
@ -14,7 +14,7 @@ Legend:
|
||||||
|
|
||||||
| Operation | BLAS | CPU | CUDA | Metal | SYCL | Vulkan |
|
| Operation | BLAS | CPU | CUDA | Metal | SYCL | Vulkan |
|
||||||
|-----------|------|------|------|------|------|------|
|
|-----------|------|------|------|------|------|------|
|
||||||
| ABS | ❌ | ✅ | 🟡 | ❌ | 🟡 | ❌ |
|
| ABS | ❌ | ✅ | 🟡 | 🟡 | 🟡 | ❌ |
|
||||||
| ACC | ❌ | ✅ | ✅ | ✅ | ✅ | ✅ |
|
| ACC | ❌ | ✅ | ✅ | ✅ | ✅ | ✅ |
|
||||||
| ADD | ❌ | ✅ | ✅ | 🟡 | ✅ | ✅ |
|
| ADD | ❌ | ✅ | ✅ | 🟡 | ✅ | ✅ |
|
||||||
| ADD1 | ❌ | ✅ | ✅ | ❌ | ✅ | ❌ |
|
| ADD1 | ❌ | ✅ | ✅ | ❌ | ✅ | ❌ |
|
||||||
|
@ -37,7 +37,7 @@ Legend:
|
||||||
| DIV | ❌ | ✅ | ✅ | 🟡 | ✅ | ✅ |
|
| DIV | ❌ | ✅ | ✅ | 🟡 | ✅ | ✅ |
|
||||||
| DUP | ❌ | ✅ | 🟡 | 🟡 | ✅ | 🟡 |
|
| DUP | ❌ | ✅ | 🟡 | 🟡 | ✅ | 🟡 |
|
||||||
| ELU | ❌ | ✅ | 🟡 | 🟡 | 🟡 | ❌ |
|
| ELU | ❌ | ✅ | 🟡 | 🟡 | 🟡 | ❌ |
|
||||||
| EXP | ❌ | ✅ | 🟡 | ❌ | 🟡 | ❌ |
|
| EXP | ❌ | ✅ | 🟡 | 🟡 | 🟡 | ❌ |
|
||||||
| FLASH_ATTN_EXT | ❌ | ✅ | 🟡 | 🟡 | ❌ | 🟡 |
|
| FLASH_ATTN_EXT | ❌ | ✅ | 🟡 | 🟡 | ❌ | 🟡 |
|
||||||
| GATED_LINEAR_ATTN | ❌ | ✅ | ✅ | ❌ | ✅ | ❌ |
|
| GATED_LINEAR_ATTN | ❌ | ✅ | ✅ | ❌ | ✅ | ❌ |
|
||||||
| GEGLU | ❌ | ✅ | ✅ | 🟡 | ✅ | 🟡 |
|
| GEGLU | ❌ | ✅ | ✅ | 🟡 | ✅ | 🟡 |
|
||||||
|
@ -49,8 +49,8 @@ Legend:
|
||||||
| GET_ROWS | ❌ | ✅ | 🟡 | ✅ | 🟡 | 🟡 |
|
| GET_ROWS | ❌ | ✅ | 🟡 | ✅ | 🟡 | 🟡 |
|
||||||
| GET_ROWS_BACK | ❌ | 🟡 | 🟡 | ❌ | ❌ | ❌ |
|
| GET_ROWS_BACK | ❌ | 🟡 | 🟡 | ❌ | ❌ | ❌ |
|
||||||
| GROUP_NORM | ❌ | ✅ | ✅ | ✅ | ✅ | ✅ |
|
| GROUP_NORM | ❌ | ✅ | ✅ | ✅ | ✅ | ✅ |
|
||||||
| HARDSIGMOID | ❌ | ✅ | 🟡 | ❌ | 🟡 | ❌ |
|
| HARDSIGMOID | ❌ | ✅ | 🟡 | 🟡 | 🟡 | ❌ |
|
||||||
| HARDSWISH | ❌ | ✅ | 🟡 | ❌ | 🟡 | ❌ |
|
| HARDSWISH | ❌ | ✅ | 🟡 | 🟡 | 🟡 | ❌ |
|
||||||
| IM2COL | ❌ | ✅ | ✅ | 🟡 | ✅ | ✅ |
|
| IM2COL | ❌ | ✅ | ✅ | 🟡 | ✅ | ✅ |
|
||||||
| L2_NORM | ❌ | ✅ | ✅ | ✅ | ✅ | ✅ |
|
| L2_NORM | ❌ | ✅ | ✅ | ✅ | ✅ | ✅ |
|
||||||
| LEAKY_RELU | ❌ | ✅ | ✅ | ✅ | ✅ | ✅ |
|
| LEAKY_RELU | ❌ | ✅ | ✅ | ✅ | ✅ | ✅ |
|
||||||
|
@ -72,8 +72,8 @@ Legend:
|
||||||
| REPEAT_BACK | ❌ | ✅ | ✅ | ❌ | ❌ | ✅ |
|
| REPEAT_BACK | ❌ | ✅ | ✅ | ❌ | ❌ | ✅ |
|
||||||
| RMS_NORM | ❌ | ✅ | ✅ | 🟡 | ✅ | ✅ |
|
| RMS_NORM | ❌ | ✅ | ✅ | 🟡 | ✅ | ✅ |
|
||||||
| RMS_NORM_BACK | ❌ | ✅ | ✅ | ❌ | ❌ | ✅ |
|
| RMS_NORM_BACK | ❌ | ✅ | ✅ | ❌ | ❌ | ✅ |
|
||||||
| RMS_NORM_MUL | ❌ | ❌ | ❌ | ✅ | ❌ | ❌ |
|
| RMS_NORM_MUL | ❌ | ❌ | ❌ | ❌ | ❌ | ❌ |
|
||||||
| RMS_NORM_MUL_ADD | ❌ | ✅ | ✅ | ❌ | ✅ | ✅ |
|
| RMS_NORM_MUL_ADD | ❌ | ✅ | ✅ | ✅ | ✅ | ✅ |
|
||||||
| ROLL | ❌ | ✅ | ❌ | ❌ | ❌ | ✅ |
|
| ROLL | ❌ | ✅ | ❌ | ❌ | ❌ | ✅ |
|
||||||
| ROPE | ❌ | ✅ | ✅ | ✅ | ✅ | ✅ |
|
| ROPE | ❌ | ✅ | ✅ | ✅ | ✅ | ✅ |
|
||||||
| ROPE_BACK | ❌ | ✅ | ✅ | ❌ | ❌ | ✅ |
|
| ROPE_BACK | ❌ | ✅ | ✅ | ❌ | ❌ | ✅ |
|
||||||
|
@ -82,7 +82,7 @@ Legend:
|
||||||
| SCALE | ❌ | ✅ | ✅ | ✅ | ✅ | ✅ |
|
| SCALE | ❌ | ✅ | ✅ | ✅ | ✅ | ✅ |
|
||||||
| SET | ❌ | ✅ | ❌ | ✅ | ❌ | ❌ |
|
| SET | ❌ | ✅ | ❌ | ✅ | ❌ | ❌ |
|
||||||
| SET_ROWS | ❌ | 🟡 | 🟡 | 🟡 | 🟡 | 🟡 |
|
| SET_ROWS | ❌ | 🟡 | 🟡 | 🟡 | 🟡 | 🟡 |
|
||||||
| SGN | ❌ | ✅ | 🟡 | ❌ | 🟡 | ❌ |
|
| SGN | ❌ | ✅ | 🟡 | 🟡 | 🟡 | ❌ |
|
||||||
| SIGMOID | ❌ | ✅ | 🟡 | 🟡 | 🟡 | 🟡 |
|
| SIGMOID | ❌ | ✅ | 🟡 | 🟡 | 🟡 | 🟡 |
|
||||||
| SILU | ❌ | ✅ | 🟡 | 🟡 | 🟡 | 🟡 |
|
| SILU | ❌ | ✅ | 🟡 | 🟡 | 🟡 | 🟡 |
|
||||||
| SILU_BACK | ❌ | ✅ | ✅ | ❌ | ❌ | ✅ |
|
| SILU_BACK | ❌ | ✅ | ✅ | ❌ | ❌ | ✅ |
|
||||||
|
@ -93,7 +93,7 @@ Legend:
|
||||||
| SQRT | ❌ | ✅ | ✅ | 🟡 | ✅ | ❌ |
|
| SQRT | ❌ | ✅ | ✅ | 🟡 | ✅ | ❌ |
|
||||||
| SSM_CONV | ❌ | ✅ | ✅ | ✅ | ❌ | ❌ |
|
| SSM_CONV | ❌ | ✅ | ✅ | ✅ | ❌ | ❌ |
|
||||||
| SSM_SCAN | ❌ | ✅ | ✅ | ✅ | ❌ | ❌ |
|
| SSM_SCAN | ❌ | ✅ | ✅ | ✅ | ❌ | ❌ |
|
||||||
| STEP | ❌ | ✅ | 🟡 | ❌ | 🟡 | ❌ |
|
| STEP | ❌ | ✅ | 🟡 | 🟡 | 🟡 | ❌ |
|
||||||
| SUB | ❌ | ✅ | ✅ | 🟡 | ✅ | ✅ |
|
| SUB | ❌ | ✅ | ✅ | 🟡 | ✅ | ✅ |
|
||||||
| SUM | ❌ | ✅ | ✅ | ❌ | ✅ | ✅ |
|
| SUM | ❌ | ✅ | ✅ | ❌ | ✅ | ✅ |
|
||||||
| SUM_ROWS | ❌ | ✅ | ✅ | ✅ | ✅ | ✅ |
|
| SUM_ROWS | ❌ | ✅ | ✅ | ✅ | ✅ | ✅ |
|
||||||
|
|
22799
docs/ops/Metal.csv
22799
docs/ops/Metal.csv
File diff suppressed because it is too large
Load Diff
Loading…
Reference in New Issue