Skip to content

Commit 2481241

Browse files
mdboomrwgk
andauthored
Update cuda_core/cuda/core/_memory/_managed_memory_resource.pyx
Co-authored-by: Ralf W. Grosse-Kunstleve <rwgkio@gmail.com>
1 parent a3349f1 commit 2481241

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

cuda_core/cuda/core/_memory/_managed_memory_resource.pyx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,7 @@ cdef class ManagedMemoryResource(_MemPool):
118118
if self._pref_loc_type is None:
119119
return None
120120
if self._pref_loc_type == "host":
121-
return ("host", None)
121+
return (ManagedMemoryLocationType.HOST, None)
122122
return (ManagedMemoryLocationType(self._pref_loc_type), self._pref_loc_id)
123123

124124
@property

0 commit comments

Comments
 (0)