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

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