Skip to content

Commit 127942b

Browse files
committed
Add fix
1 parent a1bb524 commit 127942b

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

cuda_bindings/cuda/bindings/_lib/utils.pxi.in

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,11 +6,11 @@ from libc.stdlib cimport calloc, free
66
from libc.stdint cimport int32_t, uint32_t, int64_t, uint64_t
77
from libc.stddef cimport wchar_t
88
from libc.string cimport memcpy
9-
from enum import Enum as _Enum
109
import ctypes as _ctypes
1110
cimport cuda.bindings.cydriver as cydriver
1211
import cuda.bindings.driver as _driver
1312
cimport cuda.bindings._lib.param_packer as param_packer
13+
from cuda.bindings._internal._fast_enum import FastEnum as _FastEnum
1414

1515
cdef void* _callocWrapper(length, size):
1616
cdef void* out = calloc(length, size)
@@ -74,7 +74,7 @@ cdef class _HelperKernelParams:
7474
self._ckernelParams[idx] = <void*><void_ptr>value.getPtr()
7575
elif isinstance(value, (_ctypes.Structure)):
7676
self._ckernelParams[idx] = <void*><void_ptr>_ctypes.addressof(value)
77-
elif isinstance(value, (_Enum)):
77+
elif isinstance(value, (_FastEnum)):
7878
self._ckernelParams[idx] = &(self._ckernelParamsData[data_idx])
7979
(<int*>self._ckernelParams[idx])[0] = value.value
8080
data_idx += sizeof(int)

0 commit comments

Comments
 (0)