17 lines
498 B
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
|