steve
/
factor
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

cuda: Allow memory to report as >4gb.

modern-harvey2
Doug Coleman 2017-08-25 19:53:13 +00:00
parent 5cc7023f16
commit 12816f4d4e
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
extra/cuda/devices/devices.factor
View File

@ -37,8 +37,8 @@ IN: cuda.devices
2array ;
: cuda-device-memory ( n -- bytes )
[ { uint } ] dip
'[ _ cuDeviceTotalMem cuda-error ] with-out-parameters ;
[ { ulonglong } ] dip
'[ _ cuDeviceTotalMem_v2 cuda-error ] with-out-parameters ;
: cuda-device-attribute ( attribute n -- n )
[ { int } ] 2dip

1
extra/cuda/ffi/ffi.factor
View File

@ -308,6 +308,7 @@ FUNCTION: CUresult cuDeviceGetCount ( int* count )
FUNCTION: CUresult cuDeviceGetName ( char* name, int len, CUdevice dev )
FUNCTION: CUresult cuDeviceComputeCapability ( int* major, int* minor, CUdevice dev )
FUNCTION: CUresult cuDeviceTotalMem ( uint* bytes, CUdevice dev )
FUNCTION: CUresult cuDeviceTotalMem_v2 ( ulonglong* bytes, CUdevice dev )
FUNCTION: CUresult cuDeviceGetProperties ( CUdevprop* prop, CUdevice dev )
FUNCTION: CUresult cuDeviceGetAttribute ( int* pi, CUdevice_attribute attrib, CUdevice dev )