Skip to content

Commit d9c2744

Browse files
EdCauntmloubout
authored andcommitted
misc: Fix oversight
1 parent ef0ef47 commit d9c2744

1 file changed

Lines changed: 6 additions & 3 deletions

File tree

devito/operator/operator.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1408,16 +1408,19 @@ def _visible_devices(self):
14081408
# Environment variable not set
14091409
continue
14101410

1411-
return {}
1412-
1411+
return None
1412+
14131413
@cached_property
14141414
def _physical_deviceid(self):
14151415
if isinstance(self.platform, Device):
14161416
# Get the physical device ID (as CUDA_VISIBLE_DEVICES may be set)
14171417
rank = self.comm.Get_rank() if self.comm != MPI.COMM_NULL else 0
14181418

14191419
logical_deviceid = max(self.get('deviceid', 0), 0) + rank
1420-
return self._visible_devices.get(logical_deviceid, logical_deviceid)
1420+
if self._visible_devices is None:
1421+
return logical_deviceid
1422+
else:
1423+
return self._visible_devices[logical_deviceid]
14211424
else:
14221425
return None
14231426

0 commit comments

Comments
 (0)