diff --git a/README.rst b/README.rst index d82c1f6da8..73f14b5e2f 100644 --- a/README.rst +++ b/README.rst @@ -8,6 +8,51 @@ **TransformerEngine-FL is a fork of TransformerEngine that introduces a plugin-based architecture for supporting diverse AI chips, built on top of** `FlagOS `_, **a unified open-source AI system software stack.** +The following table summarizes the specific adaptation status of multiple chips to the three backends of TransformerEngine-FL (FlagOS Triton, Vendor, and Reference Torch). + +.. list-table:: + :header-rows: 1 + :widths: 28 18 20 18 + :align: left + + * - Vendor + - FlagOS (Triton) + - Reference (Torch) + - Vendor + * - NVIDIA GPU + - ✅ + - ✅ + - ✅ + * - Metax + - ✅ + - ✅ + - ✅ + * - Moore Threads (MUSA) + - ✅ + - ✅ + - ✅ + * - Hygon + - ✅ + - ✅ + - ✅ + * - Iluvatar + - ✅ + - ✅ + - ✅ + * - Kunlunxin + - + - ✅ + - — + * - Cambricon + - + - ✅ + - — + * - Tsingmicro (TXDA) + - ✅ + - ✅ + - — + + Transformer Engine ==================