Skip to content

Commit a758a58

Browse files
committed
arch: fix gpu detection with latest rocm-smi
1 parent 06d3837 commit a758a58

2 files changed

Lines changed: 2 additions & 4 deletions

File tree

devito/arch/archinfo.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -270,7 +270,7 @@ def cbk(deviceid=0):
270270
for line in lines:
271271
if 'GPU' in line:
272272
# Product
273-
pattern = r'GPU\[(\d+)\].*?Card series:\s*(.*?)\s*$'
273+
pattern = r'GPU\[(\d+)\].*?Card [sS]eries:\s*(.*?)\s*$'
274274
match1 = re.match(pattern, line)
275275

276276
if match1:
@@ -280,7 +280,7 @@ def cbk(deviceid=0):
280280
gpu_infos[gid]['product'] = match1.group(2)
281281

282282
# Model
283-
pattern = r'GPU\[(\d+)\].*?Card model:\s*(.*?)\s*$'
283+
pattern = r'GPU\[(\d+)\].*?Card [mM]odel:\s*(.*?)\s*$'
284284
match2 = re.match(pattern, line)
285285

286286
if match2:

docker/Dockerfile.amd

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -146,7 +146,6 @@ FROM sdk-base as amdclang
146146
# MPI env
147147
ENV OMPI_CC="amdclang"
148148
ENV OMPI_CXX="amdclang++"
149-
ENV MPICC="amdclang"
150149

151150
# Devito env
152151
ENV DEVITO_ARCH="aomp"
@@ -162,7 +161,6 @@ FROM sdk-base as hip
162161
# MPI env
163162
ENV OMPI_CC="hipcc"
164163
ENV OMPI_CXX="hipcc"
165-
ENV MPICC="MPICC"
166164

167165
# Devito env
168166
ENV DEVITO_ARCH="hip"

0 commit comments

Comments
 (0)