diff --git a/extra/cuda/utils/utils.factor b/extra/cuda/utils/utils.factor index eef205992f..a85a0b35e9 100644 --- a/extra/cuda/utils/utils.factor +++ b/extra/cuda/utils/utils.factor @@ -92,3 +92,5 @@ ERROR: throw-cuda-error n ; : function-shared-size ( n -- ) [ cuda-function get ] dip cuFuncSetSharedSize cuda-error ; + +: distribute-jobs ( job-count per-job-shared -- grid-size block-size per-block-shared )