Commit eab2f39d authored by Kyle Chen's avatar Kyle Chen Committed by Facebook GitHub Bot
Browse files

[ROCm] Update to rocm5.1.1 (#2362)

Summary:
previous update for rocm: https://github.com/pytorch/audio/pull/2186

Pull Request resolved: https://github.com/pytorch/audio/pull/2362

Reviewed By: seemethere

Differential Revision: D36283672

Pulled By: mthrok

fbshipit-source-id: bfd38940d027c8ccd72ab48991e5ab7f84b0e9c0
parent 4b021ae3
This diff is collapsed.
...@@ -23,7 +23,7 @@ from jinja2 import select_autoescape ...@@ -23,7 +23,7 @@ from jinja2 import select_autoescape
PYTHON_VERSIONS = ["3.7", "3.8", "3.9", "3.10"] PYTHON_VERSIONS = ["3.7", "3.8", "3.9", "3.10"]
CU_VERSIONS_DICT = { CU_VERSIONS_DICT = {
"linux": ["cpu", "cu102", "cu113", "cu116", "rocm4.3.1", "rocm4.5.2"], "linux": ["cpu", "cu102", "cu113", "cu116", "rocm5.0", "rocm5.1.1"],
"windows": ["cpu", "cu113", "cu116"], "windows": ["cpu", "cu113", "cu116"],
"macos": ["cpu"], "macos": ["cpu"],
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment