From 3eba1dd27e485b386d75a264cdfe7000be639065 Mon Sep 17 00:00:00 2001 From: Stas Bekman Date: Wed, 1 Mar 2023 08:52:49 -0800 Subject: [PATCH] [doc] deepspeed tests (#21859) --- docs/source/en/main_classes/deepspeed.mdx | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/docs/source/en/main_classes/deepspeed.mdx b/docs/source/en/main_classes/deepspeed.mdx index 73c29cd405..1a5358c807 100644 --- a/docs/source/en/main_classes/deepspeed.mdx +++ b/docs/source/en/main_classes/deepspeed.mdx @@ -2245,6 +2245,24 @@ rank1: This was a very basic example and you will want to adapt it to your needs. +## Testing Deepspeed Integration + +If you submit a PR that involves DeepSpeed integration please note our CircleCI PR CI setup has no GPUs, so we only run tests requiring gpus on a different CI nightly. Therefore if you get a green CI report in your PR it doesn't mean DeepSpeed tests pass. + +To run DeepSpeed tests, please run at least: + +``` +RUN_SLOW=1 pytest tests/deepspeed/test_deepspeed.py +``` + +If you changed any of the modeling or pytorch examples code, then run the model zoo tests as well. The following will run all DeepSpeed tests: + +``` +RUN_SLOW=1 pytest tests/deepspeed +``` + + + ## Main DeepSpeed Resources