Platforms Supported#
FlagGems supports 10+ backends. The currently supported platforms and their capabilities are listed blow:
Vendor |
Vendor name |
State |
float16 |
float32 |
bfloat16 |
|---|---|---|---|---|---|
AIPU |
|
☑️ |
✅ |
✅ |
✅ |
AMD |
|
🚧 |
- |
- |
- |
ARM(CPU) |
|
☑️ |
- |
- |
- |
Ascend |
|
✅ |
✅ |
✅ |
✅ |
Cambricon |
|
✅ |
✅ |
✅ |
✅ |
Hygon |
|
✅ |
✅ |
✅ |
✅ |
Iluvatar |
|
✅ |
✅ |
✅ |
✅ |
Kunlunxin |
|
✅ |
✅ |
✅ |
✅ |
MetaX |
|
✅ |
✅ |
✅ |
✅ |
Mthreads |
|
✅ |
✅ |
✅ |
✅ |
NVIDIA |
|
✅ |
✅ |
✅ |
✅ |
Sunrise |
|
☑️ |
✅ |
✅ |
✅ |
TsingMicro |
|
🚧 |
- |
- |
- |
Legend:
✅ - supported
☑️ - partially supported
🚧 - under development