extra/cuda/nvcc/nvcc.factor

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