cuda.demos.prefix-sum: add init-cuda

2010-05-15 15:01:32 -07:00 · 2010-05-15 15:01:32 -07:00 · 896cd58451
parent 3de020c18a
commit 896cd58451
1 changed files with 1 additions and 0 deletions
--- a/extra/cuda/demos/prefix-sum/prefix-sum.factor
+++ b/extra/cuda/demos/prefix-sum/prefix-sum.factor
@ -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-prefix-sum ( -- )
+    init-cuda
    0 0 [
        ! { 1 1 1 } { 2 1 } 0 3<<< prefix_sum_block
    ] with-cuda-context ;