Skip to content

Commit b879fa5

Browse files
committed
fixing ci compiler errors
1 parent c418050 commit b879fa5

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

cuda_core/cuda/core/_memory/_buffer.pyx

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ else:
3636
BufferProtocol = object
3737

3838
from cuda.core._dlpack import DLDeviceType, make_py_capsule
39-
from cuda.core._utils.cuda_utils import driver
39+
from cuda.core._utils.cuda_utils import driver, handle_return
4040
from cuda.core._device import Device
4141

4242

@@ -175,7 +175,7 @@ cdef inline object _normalize_managed_location(
175175
cdef int loc_id
176176

177177
if isinstance(location, Device):
178-
location = (<Device>location).device_id
178+
location = location.device_id
179179

180180
if location_type is not None and not isinstance(location_type, str):
181181
raise TypeError(f"{what} location_type must be a string or None, got {type(location_type).__name__}")

0 commit comments

Comments
 (0)