cuda.devices: fix distribute-jobs to look only at width of maxThreadsDim
parent
6d41ea32f5
commit
077cd0397b
|
@ -82,5 +82,5 @@ IN: cuda.devices
|
||||||
|
|
||||||
: distribute-jobs ( job-count per-job-shared -- launcher )
|
: distribute-jobs ( job-count per-job-shared -- launcher )
|
||||||
cuda-device get cuda-device-properties
|
cuda-device get cuda-device-properties
|
||||||
[ sharedMemPerBlock>> ] [ maxThreadsDim>> ] bi
|
[ sharedMemPerBlock>> ] [ maxThreadsDim>> first ] bi
|
||||||
(distribute-jobs) 3<<< ; inline
|
(distribute-jobs) 3<<< ; inline
|
||||||
|
|
Loading…
Reference in New Issue