factor/extra/cuda/demos/prefix-sum/prefix-sum.factor

17 lines
498 B
Factor

! Copyright (C) 2010 Doug Coleman.
! See http://factorcode.org/license.txt for BSD license.
USING: alien.c-types cuda cuda.contexts cuda.libraries cuda.syntax locals ;
IN: cuda.demos.prefix-sum
CUDA-LIBRARY: prefix-sum cuda32 "vocab:cuda/demos/prefix-sum/prefix-sum.ptx"
CUDA-FUNCTION: prefix_sum_block ( uint* in, uint* out, uint n )
:: cuda-prefix-sum ( -- )
init-cuda
0 0 [
! { 1 1 1 } { 2 1 } 0 <grid-shared> prefix_sum_block
] with-cuda-context ;
MAIN: cuda-prefix-sum