cuda.demos.prefix-sum: add init-cuda
parent
3de020c18a
commit
896cd58451
|
|
@ -8,6 +8,7 @@ CUDA-LIBRARY: prefix-sum vocab:cuda/demos/prefix-sum/prefix-sum.ptx
|
||||||
CUDA-FUNCTION: prefix_sum_block ( uint* in, uint* out, uint n ) ;
|
CUDA-FUNCTION: prefix_sum_block ( uint* in, uint* out, uint n ) ;
|
||||||
|
|
||||||
:: cuda-prefix-sum ( -- )
|
:: cuda-prefix-sum ( -- )
|
||||||
|
init-cuda
|
||||||
0 0 [
|
0 0 [
|
||||||
! { 1 1 1 } { 2 1 } 0 3<<< prefix_sum_block
|
! { 1 1 1 } { 2 1 } 0 3<<< prefix_sum_block
|
||||||
] with-cuda-context ;
|
] with-cuda-context ;
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue