steve
/
factor
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

cuda.libraries, cuda.syntax: factor generated CUDA-FUNCTION: word bodies into a "cuda-invoke" macro; add CUDA-GLOBAL: for referencing global symbols

db4
Joe Groff 2010-05-20 15:45:35 -07:00
parent f0a7f039bd
commit efe7e94a07
2 changed files with 25 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
extra/cuda/libraries/libraries.factor
View File

@ -155,17 +155,29 @@ MACRO: cuda-arguments ( c-types -- quot: ( args... function -- ) )
[ cached-module ] dip
2array cuda-functions get [ first2 get-function-ptr* ] cache ;
: define-cuda-word ( word module-name function-name arguments -- )
[
'[
_ _ cached-function
[ nip _ cuda-arguments ]
[ run-grid ] 2bi
]
]
MACRO: cuda-invoke ( module-name function-name arguments -- )
'[
_ _ cached-function
[ nip _ cuda-arguments ]
[ run-grid ] 2bi
] ;
: cuda-global* ( module-name symbol-name -- device-ptr size )
[ CUdeviceptr <c-object> c:uint <c-object> ] 2dip
[ cached-module ] dip
'[ _ _ cuModuleGetGlobal cuda-error ] 2keep [ c:*uint ] bi@ ; inline
: cuda-global ( module-name symbol-name -- device-ptr )
cuda-global* drop ; inline
: define-cuda-function ( word module-name function-name arguments -- )
[ '[ _ _ _ cuda-invoke ] ]
[ 2nip \ grid suffix c:void function-effect ]
3bi define-declared ;
: define-cuda-global ( word module-name symbol-name -- )
'[ _ _ cuda-global ] (( -- device-ptr )) define-declared ;
TUPLE: cuda-library name path handle ;
: <cuda-library> ( name path -- obj )

7
extra/cuda/syntax/syntax.factor
View File

@ -10,6 +10,9 @@ SYNTAX: CUDA-LIBRARY:
[ drop current-cuda-library set-global ] 2bi ;
SYNTAX: CUDA-FUNCTION:
scan [ create-in current-cuda-library get ] [ ] bi
";" scan-c-args drop define-cuda-word ;
scan [ create-in current-cuda-library get ] keep
";" scan-c-args drop define-cuda-function ;
SYNTAX: CUDA-GLOBAL:
scan [ create-in current-cuda-library get ] keep
define-cuda-global ;