[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
4148591fa7
commit
ae0f9afa24
1 changed files with 1 additions and 0 deletions
|
|
@ -72,6 +72,7 @@ def _detect_rocm_version() -> tuple[int, int] | None:
|
|||
)
|
||||
if result.returncode == 0:
|
||||
import re
|
||||
|
||||
m = re.search(r"ROCm version:\s*(\d+)\.(\d+)", result.stdout)
|
||||
if m:
|
||||
return int(m.group(1)), int(m.group(2))
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue