init-cuda in a couple more places...

db4
Doug Coleman 2010-04-23 23:19:26 -05:00
parent 8d76f2e612
commit cc546c3a94
1 changed files with 4 additions and 2 deletions

View File

@ -20,7 +20,8 @@ IN: cuda.devices
: with-each-cuda-device ( quot -- )
[ enumerate-cuda-devices ] dip '[ <launcher> _ with-cuda ] each ; inline
: cuda-device-properties ( device -- properties )
: cuda-device-properties ( n -- properties )
init-cuda
[ CUdevprop <c-object> ] 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 <c-object> ] 2dip
[ cuDeviceGetAttribute cuda-error ]
[ 2drop *int ] 3bi ;