We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent c418050 commit b879fa5Copy full SHA for b879fa5
cuda_core/cuda/core/_memory/_buffer.pyx
@@ -36,7 +36,7 @@ else:
36
BufferProtocol = object
37
38
from cuda.core._dlpack import DLDeviceType, make_py_capsule
39
-from cuda.core._utils.cuda_utils import driver
+from cuda.core._utils.cuda_utils import driver, handle_return
40
from cuda.core._device import Device
41
42
@@ -175,7 +175,7 @@ cdef inline object _normalize_managed_location(
175
cdef int loc_id
176
177
if isinstance(location, Device):
178
- location = (<Device>location).device_id
+ location = location.device_id
179
180
if location_type is not None and not isinstance(location_type, str):
181
raise TypeError(f"{what} location_type must be a string or None, got {type(location_type).__name__}")
0 commit comments