From 338e571d6af2b2ea52f29a70ba4965c6e7f54332 Mon Sep 17 00:00:00 2001 From: Logan Adams Date: Tue, 17 Dec 2024 14:59:09 -0800 Subject: [PATCH] Test with setting ds_accelerator explicitly --- .github/workflows/no-torch.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/no-torch.yml b/.github/workflows/no-torch.yml index 1a13c0f3f4f1..b26b4efd9203 100644 --- a/.github/workflows/no-torch.yml +++ b/.github/workflows/no-torch.yml @@ -35,7 +35,7 @@ jobs: - name: Build deepspeed run: | - DS_BUILD_STRING=" " python setup.py sdist + DS_ACCELERATOR="cuda" DS_BUILD_STRING=" " python setup.py sdist - name: Open GitHub issue if nightly CI fails if: ${{ failure() && (github.event_name == 'schedule') }}