cuda.devices: fix distribute-jobs to look only at width of maxThreadsDim

diff --git a/extra/cuda/devices/devices.factor b/extra/cuda/devices/devices.factor
index d909e9ab5c753252376c6c4c54ce40a0be279f3f..7ad7b32c8d5e5d9563dce96cd7dbb900957a9c71 100644 (file)
--- a/extra/cuda/devices/devices.factor
+++ b/extra/cuda/devices/devices.factor
@@ -82,5 +82,5 @@ IN: cuda.devices
 
 : distribute-jobs ( job-count per-job-shared -- launcher )
     cuda-device get cuda-device-properties 
-    [ sharedMemPerBlock>> ] [ maxThreadsDim>> ] bi
+    [ sharedMemPerBlock>> ] [ maxThreadsDim>> first ] bi
     (distribute-jobs) 3<<< ; inline