1 ! Copyright (C) 2010 Doug Coleman.
2 ! See http://factorcode.org/license.txt for BSD license.
3 USING: arrays combinators.smart io.backend io.directories
4 io.launcher io.pathnames kernel locals math sequences splitting
8 HOOK: nvcc-path os ( -- path )
10 M: object nvcc-path "nvcc" ;
12 M: macosx nvcc-path "/usr/local/cuda/bin/nvcc" ;
14 : cu>ptx ( path -- path' )
15 ".cu" ?tail drop ".ptx" append ;
17 : nvcc-command ( path -- seq )
19 [ nvcc-path "--ptx" "-o" ] dip
20 [ cu>ptx ] [ file-name ] bi
23 ERROR: nvcc-failed n path ;
25 :: compile-cu ( path -- path' )
26 path normalize-path :> path2
27 path2 parent-directory [
29 run-process wait-for-process [ path2 nvcc-failed ] unless-zero