diff --git a/extra/cuda/devices/devices.factor b/extra/cuda/devices/devices.factor index c9117bd00c..8b29295a0b 100644 --- a/extra/cuda/devices/devices.factor +++ b/extra/cuda/devices/devices.factor @@ -20,7 +20,8 @@ IN: cuda.devices : with-each-cuda-device ( quot -- ) [ enumerate-cuda-devices ] dip '[ _ with-cuda ] each ; inline -: cuda-device-properties ( device -- properties ) +: cuda-device-properties ( n -- properties ) + init-cuda [ CUdevprop ] dip [ cuDeviceGetProperties cuda-error ] 2keep drop CUdevprop memory>struct ; @@ -46,7 +47,8 @@ IN: cuda.devices [ cuDeviceTotalMem cuda-error ] [ drop *uint ] 2bi ; -: cuda-device-attribute ( attribute dev -- n ) +: cuda-device-attribute ( attribute n -- n ) + init-cuda [ int ] 2dip [ cuDeviceGetAttribute cuda-error ] [ 2drop *int ] 3bi ;