mirror of https://gitlab.com/ita1024/waf.git
64 lines
1.6 KiB
Python
64 lines
1.6 KiB
Python
#!/usr/bin/env python
|
|
# encoding: utf-8
|
|
# Thomas Nagy, 2010
|
|
|
|
"cuda"
|
|
|
|
from waflib import Task
|
|
from waflib.TaskGen import extension
|
|
from waflib.Tools import ccroot, c_preproc
|
|
from waflib.Configure import conf
|
|
|
|
class cuda(Task.Task):
|
|
run_str = '${NVCC} ${CUDAFLAGS} ${NVCCFLAGS_ST:CXXFLAGS} ${FRAMEWORKPATH_ST:FRAMEWORKPATH} ${CPPPATH_ST:INCPATHS} ${DEFINES_ST:DEFINES} ${CXX_SRC_F}${SRC} ${CXX_TGT_F} ${TGT}'
|
|
color = 'GREEN'
|
|
ext_in = ['.h']
|
|
vars = ['CCDEPS']
|
|
scan = c_preproc.scan
|
|
shell = False
|
|
|
|
@extension('.cu', '.cuda')
|
|
def c_hook(self, node):
|
|
return self.create_compiled_task('cuda', node)
|
|
|
|
@extension('.cpp')
|
|
def cxx_hook(self, node):
|
|
# override processing for one particular type of file
|
|
if getattr(self, 'cuda', False):
|
|
return self.create_compiled_task('cuda', node)
|
|
else:
|
|
return self.create_compiled_task('cxx', node)
|
|
|
|
def configure(conf):
|
|
conf.find_program('nvcc', var='NVCC')
|
|
conf.find_cuda_libs()
|
|
conf.env.NVCCFLAGS_ST = "--compiler-options=%s"
|
|
|
|
@conf
|
|
def find_cuda_libs(self):
|
|
"""
|
|
find the cuda include and library folders
|
|
|
|
use ctx.program(source='main.c', target='app', use='CUDA CUDART')
|
|
"""
|
|
|
|
if not self.env.NVCC:
|
|
self.fatal('check for nvcc first')
|
|
|
|
d = self.root.find_node(self.env.NVCC[0]).parent.parent
|
|
|
|
node = d.find_node('include')
|
|
_includes = node and node.abspath() or ''
|
|
|
|
_libpath = []
|
|
for x in ('lib64', 'lib64/stubs', 'lib', 'lib/stubs'):
|
|
try:
|
|
_libpath.append(d.find_node(x).abspath())
|
|
except:
|
|
pass
|
|
|
|
# this should not raise any error
|
|
self.check_cxx(header='cuda.h', lib='cuda', libpath=_libpath, includes=_includes)
|
|
self.check_cxx(header='cuda.h', lib='cudart', libpath=_libpath, includes=_includes)
|
|
|