diff --git a/extra/cuda/devices/devices.factor b/extra/cuda/devices/devices.factor index 8b29295a0b..e5f72f50d6 100644 --- a/extra/cuda/devices/devices.factor +++ b/extra/cuda/devices/devices.factor @@ -7,11 +7,9 @@ sequences ; IN: cuda.devices : #cuda-devices ( -- n ) - init-cuda int [ cuDeviceGetCount cuda-error ] keep *int ; : n>cuda-device ( n -- device ) - init-cuda [ CUdevice ] dip [ cuDeviceGet cuda-error ] 2keep drop *int ; : enumerate-cuda-devices ( -- devices ) @@ -21,7 +19,6 @@ IN: cuda.devices [ enumerate-cuda-devices ] dip '[ _ with-cuda ] each ; inline : cuda-device-properties ( n -- properties ) - init-cuda [ CUdevprop ] dip [ cuDeviceGetProperties cuda-error ] 2keep drop CUdevprop memory>struct ; @@ -30,31 +27,26 @@ IN: cuda.devices enumerate-cuda-devices [ dup cuda-device-properties ] { } map>assoc ; : cuda-device-name ( n -- string ) - init-cuda [ 256 [ ] keep ] dip [ cuDeviceGetName cuda-error ] [ 2drop utf8 alien>string ] 3bi ; : cuda-device-capability ( n -- pair ) - init-cuda [ int int ] dip [ cuDeviceComputeCapability cuda-error ] [ drop [ *int ] bi@ ] 3bi 2array ; : cuda-device-memory ( n -- bytes ) - init-cuda [ uint ] dip [ cuDeviceTotalMem cuda-error ] [ drop *uint ] 2bi ; : cuda-device-attribute ( attribute n -- n ) - init-cuda [ int ] 2dip [ cuDeviceGetAttribute cuda-error ] [ 2drop *int ] 3bi ; : cuda-device. ( n -- ) - init-cuda { [ "Device: " write number>string print ] [ "Name: " write cuda-device-name print ]