Skip to content

Add torch-latest and torch-nightly CI workflows#1990

Merged
jeffra merged 15 commits intodeepspeedai:masterfrom
mrwyattii:add-torch-latest-workflow
Jun 6, 2022
Merged

Add torch-latest and torch-nightly CI workflows#1990
jeffra merged 15 commits intodeepspeedai:masterfrom
mrwyattii:add-torch-latest-workflow

Conversation

@mrwyattii
Copy link
Contributor

This PR adds a new workflow that runs unit tests on the latest stable torch release and the nightly release. The nightly is a scheduled workflow and runs once daily on main DeepSpeed branch.

Also addresses #1987 by extending torch information stored in deepspeed.git_version_info.torch_info. This was necessary to avoid initializing CUDA with torch.cuda.is_bf16_supported() before a distributed unit test when bf16_required_version_check() was used.

@jeffra jeffra merged commit 7fc3065 into deepspeedai:master Jun 6, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants