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

2017-06-03 20:59:51 -05:00 · 2017-06-03 20:59:51 -05:00 · b3ba829601
parent acffb259f7
commit b3ba829601
1 changed files with 4 additions and 0 deletions
--- a/extra/cuda/devices/devices.factor
+++ b/extra/cuda/devices/devices.factor
@ -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