cuda.devices: Utility word to print all cuda devices.

char-rename
Doug Coleman 2017-06-03 20:59:51 -05:00
parent acffb259f7
commit b3ba829601
1 changed files with 4 additions and 0 deletions

View File

@ -61,6 +61,10 @@ IN: cuda.devices
]
} cleave ;
: cuda-devices. ( -- )
init-cuda
enumerate-cuda-devices [ cuda-device. ] each ;
: cuda. ( -- )
init-cuda
"CUDA Version: " write cuda-version number>string print nl