From 3e4ad8a2f589dc33390ec042c48e5f8aaff20448 Mon Sep 17 00:00:00 2001 From: Keita Watanabe Date: Mon, 5 Aug 2024 07:25:01 +0900 Subject: [PATCH] Update README.md Update support matrix --- README.md | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/README.md b/README.md index e4eb28ae..47cfd935 100644 --- a/README.md +++ b/README.md @@ -50,20 +50,20 @@ All test cases are under `3.test_cases/`. You can go in each test case directory | Test cases | Slurm | Kubernetes | AWS Batch | | ----------------------------------------------------------------------------- | ----- | ----------- | ---------- | -| [`1.megatron-lm`](./3.test_cases/1.megatron-lm) | ✅ | ❓ | ❓ | -| [`2.nemo-launcher`](./3.test_cases/2.nemo-launcher) | ✅ | ❌ | ❌ | +| [`1.megatron-lm`](./3.test_cases/1.megatron-lm) | ✅ | ✅ | ❓ | +| [`2.nemo-launcher`](./3.test_cases/2.nemo-launcher) | ✅ | ✅ | ❌ | | [`3.MPT`](./3.test_cases/3.MPT) | ✅ | ❓ | ❓ | | [`4.DDP`](./3.test_cases/4.DDP) | ✅ | ❓ | ❓ | | [`5.param-benchmark`](./3.test_cases/5.param-benchmark) | ✅ | ❓ | ❓ | | [`6.stable-diffusion`](./3.test_cases/6.stable-diffusion) | ✅ | ✅ | ❓ | | [`7.tensorflow-distributed`](./3.test_cases/7.tensorflow-distributed) | ✅ | ❓ | ❓ | -| [`10.FSDP`](./3.test_cases/10.FSDP) | ✅ | ❓ | ❓ | +| [`10.FSDP`](./3.test_cases/10.FSDP) | ✅ | ✅ | ❓ | | [`11.modelparallel`](./3.test_cases/11.modelparallel) | ✅ | ❓ | ❓ | | [`12.SM-dataparallel-FSDP`](./3.test_cases/12.SM-dataparallel-FSDP) | ✅ | ❓ | ❓ | | [`13.SM-dataparallel-deepspeed`](./3.test_cases/13.SM-dataparallel-deepspeed) | ✅ | ❓ | ❓ | | [`14.bionemo`](./3.test_cases/14.bionemo) | ✅ | ❓ | ❓ | | [`15.gpt-neox`](./3.test_cases/15.gpt-neox) | ✅ | ❓ | ❓ | -| [`16.pytorch-cpu-ddp`](./3.test_cases/16.pytorch-cpu-ddp) | ✅ | ❓ | ❓ | +| [`16.pytorch-cpu-ddp`](./3.test_cases/16.pytorch-cpu-ddp) | ✅ | ✅ | ❓ | | [`17.SM-modelparallelv2`](./3.test_cases/17.SM-modelparallelv2) | ✅ | ❓ | ❓ | | [`18.deepspeed`](./3.test_cases/18.deepspeed) | ✅ | ❓ | ❓ | | [`jax`](./3.test_cases/jax) | ✅ | ❓ | ❓ |