Skip to content

Commit 9cd877c

Browse files
mdboomcpcloud
andauthored
Apply suggestion from @cpcloud
Co-authored-by: Phillip Cloud <417981+cpcloud@users.noreply.github.com>
1 parent 39be54d commit 9cd877c

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

cuda_core/cuda/core/system/_device.pyx

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -167,8 +167,7 @@ cdef class Device:
167167
"""
168168
arch = nvml.device_get_architecture(self._handle)
169169
try:
170-
arch = DeviceArch(arch)
171-
return arch
170+
return DeviceArch(arch)
172171
except ValueError:
173172
return nvml.DeviceArch.UNKNOWN
174173

0 commit comments

Comments
 (0)