File tree Expand file tree Collapse file tree
cuda_bindings/cuda/bindings/_lib Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -6,11 +6,11 @@ from libc.stdlib cimport calloc, free
66from libc.stdint cimport int32_t, uint32_t, int64_t, uint64_t
77from libc.stddef cimport wchar_t
88from libc.string cimport memcpy
9- from enum import Enum as _Enum
109import ctypes as _ctypes
1110cimport cuda.bindings.cydriver as cydriver
1211import cuda.bindings.driver as _driver
1312cimport cuda.bindings._lib.param_packer as param_packer
13+ from cuda.bindings._internal._fast_enum import FastEnum as _FastEnum
1414
1515cdef 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)
You can’t perform that action at this time.
0 commit comments