22 lines
539 B
Factor
22 lines
539 B
Factor
|
! Copyright (C) 2010 Doug Coleman.
|
||
|
|
! See http://factorcode.org/license.txt for BSD license.
|
||
|
|
USING: alien.c-types cuda cuda.syntax locals ;
|
||
|
|
IN: cuda.demos.prefix-sum
|
||
|
|
|
||
|
|
CUDA-LIBRARY: prefix-sum vocab:cuda/demos/prefix-sum/prefix-sum.ptx
|
||
|
|
|
||
|
|
CUDA-FUNCTION: prefix_sum_block ( uint* in, uint* out, uint n ) ;
|
||
|
|
|
||
|
|
:: cuda-prefix-sum ( -- )
|
||
|
|
T{ launcher
|
||
|
|
{ device 0 }
|
||
|
|
{ path "vocab:cuda/demos/prefix-sum/prefix-sum.ptx" }
|
||
|
|
} [
|
||
|
|
|
||
|
|
|
||
|
|
! { 1 1 1 } { 2 1 } 0 3<<< prefix_sum_block
|
||
|
|
|
||
|
|
] with-cuda ;
|
||
|
|
|
||
|
|
MAIN: cuda-prefix-sum
|