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.syntax locals ;
   4 IN: cuda.demos.prefix-sum
   5
   6 CUDA-LIBRARY: prefix-sum 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     T{ launcher { device 0 } }
  12     [
  13         ! { 1 1 1 } { 2 1 } 0 3<<< prefix_sum_block
  14     ] with-cuda ;
  15
  16 MAIN: cuda-prefix-sum