! Copyright (C) 2010 Doug Coleman.
! See http://factorcode.org/license.txt for BSD license.
USING: accessors alien.c-types alien.strings byte-arrays cuda
cuda.contexts cuda.devices cuda.libraries cuda.memory cuda.syntax
destructors io io.encodings.string io.encodings.utf8 kernel locals
math math.parser namespaces sequences strings ;
IN: cuda.demos.hello-world

CUDA-LIBRARY: hello cuda32 vocab:cuda/demos/hello-world/hello.ptx

CUDA-FUNCTION: helloWorld ( char* string-ptr ) ;

: cuda-hello-world ( -- )
    init-cuda
    [
        [
            context-device number>string
            "CUDA device " ": " surround write
            "Hello World!" >byte-array [ - ] map-index host>device &cuda-free

            [ { 2 1 } { 6 1 1 } <grid> helloWorld ]
            [ 12 device>host >string print ] bi
        ] with-destructors
    ] with-each-cuda-device ;

MAIN: cuda-hello-world