Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

ROCm and ROCm Apex fixes for AMD workflows #4230

Closed
wants to merge 10 commits into from
Closed

ROCm and ROCm Apex fixes for AMD workflows #4230

Changes from 3 commits
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
71f73d7
Update ROCm installed version to 5.5
loadams Aug 29, 2023
33a1b57
Test with MI200 workflow enabled
loadams Aug 29, 2023
b24aaac
Merge branch 'master' into loadams/rocm-fixes
loadams Aug 29, 2023
1ea0e85
Merge branch 'master' into loadams/rocm-fixes
loadams Aug 31, 2023
fb75f34
Remove PR toggle after adding workflow dispatch
loadams Aug 31, 2023
7cbd4fb
Test updating the apex version
loadams Sep 11, 2023
0f254e5
Merge branch 'loadams/amd-apex' into loadams/rocm-fixes
loadams Sep 14, 2023
5cec8fc
Merge branch 'master' into loadams/rocm-fixes
loadams Sep 14, 2023
bb5c1b7
Merge branch 'master' into loadams/rocm-fixes
loadams Oct 4, 2023
f7a9ad9
Merge branch 'master' into loadams/rocm-fixes
loadams Oct 18, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 5 additions & 1 deletion .github/workflows/amd-mi200.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,6 +1,10 @@
name: amd-mi200

on:
pull_request:
paths-ignore:
- 'docs/**'
- 'blogs/**'
schedule:
- cron: "0 0 * * *"

Expand All @@ -27,7 +31,7 @@ jobs:

- name: Install pytorch
run: |
pip install -U --cache-dir $TORCH_CACHE torch torchvision --extra-index-url https://download.pytorch.org/whl/rocm5.4.2
pip install -U --cache-dir $TORCH_CACHE torch torchvision --extra-index-url https://download.pytorch.org/whl/rocm5.5
loadams marked this conversation as resolved.
Show resolved Hide resolved
python -c "import torch; print('torch:', torch.__version__, torch)"
python -c "import torch; print('CUDA available:', torch.cuda.is_available())"

Expand Down