Updates for numeric array

Author: mdboom

cuda_bindings/cuda/bindings/cufile.pyx

@@ -16,7 +16,7 @@ import cython
 from cuda.bindings.driver import CUresult as pyCUresult
 
 from libc.stdlib cimport calloc, free, malloc
-cimport cpython
+from cython cimport view
 cimport cpython.buffer
 cimport cpython.memoryview
 from libc.string cimport memcmp, memcpy
@@ -2185,24 +2185,32 @@ cdef class StatsLevel2:
     @property
     def read_size_kb_hist(self):
         """~_numpy.uint64: (array of length 32)."""
-        return self._ptr[0].read_size_kb_hist
+        cdef view.array arr = view.array(shape=(32,), itemsize=sizeof(uint64_t), format="Q", mode="c", allocate_buffer=False)
+        arr.data = <char *>(&(self._ptr[0].read_size_kb_hist))
+        return arr
 
     @read_size_kb_hist.setter
     def read_size_kb_hist(self, val):
         if self._readonly:
             raise ValueError("This StatsLevel2 instance is read-only")
-        self._ptr[0].read_size_kb_hist = val
+        cdef view.array arr = view.array(shape=(len(val),), itemsize=sizeof(uint64_t), format="Q", mode="c")
+        arr[:] = _numpy.asarray(val, dtype=_numpy.uint64)
+        memcpy(<void *>(&(self._ptr[0].read_size_kb_hist)), <void *>(arr.data), sizeof(uint64_t) * len(val))
 
     @property
     def write_size_kb_hist(self):
         """~_numpy.uint64: (array of length 32)."""
-        return self._ptr[0].write_size_kb_hist
+        cdef view.array arr = view.array(shape=(32,), itemsize=sizeof(uint64_t), format="Q", mode="c", allocate_buffer=False)
+        arr.data = <char *>(&(self._ptr[0].write_size_kb_hist))
+        return arr
 
     @write_size_kb_hist.setter
     def write_size_kb_hist(self, val):
         if self._readonly:
             raise ValueError("This StatsLevel2 instance is read-only")
-        self._ptr[0].write_size_kb_hist = val
+        cdef view.array arr = view.array(shape=(len(val),), itemsize=sizeof(uint64_t), format="Q", mode="c")
+        arr[:] = _numpy.asarray(val, dtype=_numpy.uint64)
+        memcpy(<void *>(&(self._ptr[0].write_size_kb_hist)), <void *>(arr.data), sizeof(uint64_t) * len(val))
 
     @staticmethod
     def from_data(data):

cuda_bindings/cuda/bindings/cycufile.pxd

@@ -185,42 +185,50 @@ cdef extern from '<cufile.h>':
     ctypedef struct CUfileError_t 'CUfileError_t':
         CUfileOpError err
         CUresult cu_err
+
 cdef struct _anon_pod0 '_anon_pod0':
     unsigned int major_version
     unsigned int minor_version
     size_t poll_thresh_size
     size_t max_direct_io_size
     unsigned int dstatusflags
     unsigned int dcontrolflags
+
 cdef extern from '<cufile.h>':
     ctypedef struct cufileRDMAInfo_t 'cufileRDMAInfo_t':
         int version
         int desc_len
         char* desc_str
+
 cdef extern from '<cufile.h>':
     ctypedef struct CUfileFSOps_t 'CUfileFSOps_t':
         char* (*fs_type)(void*)
         int (*getRDMADeviceList)(void*, sockaddr_t**)
         int (*getRDMADevicePriority)(void*, char*, size_t, loff_t, sockaddr_t*)
         ssize_t (*read)(void*, char*, size_t, loff_t, cufileRDMAInfo_t*)
         ssize_t (*write)(void*, const char*, size_t, loff_t, cufileRDMAInfo_t*)
+
 cdef union _anon_pod1 '_anon_pod1':
     int fd
     void* handle
+
 cdef struct _anon_pod3 '_anon_pod3':
     void* devPtr_base
     off_t file_offset
     off_t devPtr_offset
     size_t size
+
 cdef extern from '<cufile.h>':
     ctypedef struct CUfileIOEvents_t 'CUfileIOEvents_t':
         void* cookie
         CUfileStatus_t status
         size_t ret
+
 cdef extern from '<cufile.h>':
     ctypedef struct CUfileOpCounter_t 'CUfileOpCounter_t':
         uint64_t ok
         uint64_t err
+
 cdef extern from '<cufile.h>':
     ctypedef struct CUfilePerGpuStats_t 'CUfilePerGpuStats_t':
         char uuid[16]
@@ -253,6 +261,7 @@ cdef extern from '<cufile.h>':
         uint64_t n_mmap_err
         uint64_t n_mmap_free
         uint64_t reg_bytes
+
 cdef extern from '<cufile.h>':
     ctypedef struct CUfileDrvProps_t 'CUfileDrvProps_t':
         _anon_pod0 nvfs
@@ -262,13 +271,16 @@ cdef extern from '<cufile.h>':
         unsigned int max_device_pinned_mem_size
         unsigned int max_batch_io_size
         unsigned int max_batch_io_timeout_msecs
+
 cdef extern from '<cufile.h>':
     ctypedef struct CUfileDescr_t 'CUfileDescr_t':
         CUfileFileHandleType type
         _anon_pod1 handle
         CUfileFSOps_t* fs_ops
+
 cdef union _anon_pod2 '_anon_pod2':
     _anon_pod3 batch
+
 cdef extern from '<cufile.h>':
     ctypedef struct CUfileStatsLevel1_t 'CUfileStatsLevel1_t':
         CUfileOpCounter_t read_ops
@@ -314,25 +326,29 @@ cdef extern from '<cufile.h>':
         uint64_t batch_completion_lat_sum_us
         uint64_t last_batch_read_bytes
         uint64_t last_batch_write_bytes
+
 cdef extern from '<cufile.h>':
     ctypedef struct CUfileIOParams_t 'CUfileIOParams_t':
         CUfileBatchMode_t mode
         _anon_pod2 u
         CUfileHandle_t fh
         CUfileOpcode_t opcode
         void* cookie
+
 cdef extern from '<cufile.h>':
     ctypedef struct CUfileStatsLevel2_t 'CUfileStatsLevel2_t':
         CUfileStatsLevel1_t basic
         uint64_t read_size_kb_hist[32]
         uint64_t write_size_kb_hist[32]
+
 cdef extern from '<cufile.h>':
     ctypedef struct CUfileStatsLevel3_t 'CUfileStatsLevel3_t':
         CUfileStatsLevel2_t detailed
         uint32_t num_gpus
         CUfilePerGpuStats_t per_gpu_stats[16]
 
 
+
 cdef extern from *:
     """
     // This is the missing piece we need to supply to help Cython & C++ compilers.