-
-
Notifications
You must be signed in to change notification settings - Fork 319
CudaTool
The following tool, cuda (cuda.py) is for using the CUDA Toolkit by NVidia. See the end of the page for an example SConscript for building the simpleGL sample that comes with the SDK and some notes.
Note: save this as cuda.py
#!python
"""
SCons.Tool.cuda
CUDA Tool for SCons
"""
import os
import sys
import SCons.Tool
import SCons.Scanner.C
import SCons.Defaults
CUDAScanner = SCons.Scanner.C.CScanner()
def CUDANVCCStaticObjectEmitter(target, source, env):
tgt, src = SCons.Defaults.StaticObjectEmitter(target, source, env)
for file in tgt:
lifile = os.path.splitext(file.rstr())[0] + '.linkinfo'
env.SideEffect( lifile, file )
env.Clean( file, lifile )
return tgt, src
def CUDANVCCSharedObjectEmitter(target, source, env):
tgt, src = SCons.Defaults.SharedObjectEmitter(target, source, env)
for file in tgt:
lifile = os.path.splitext(file.rstr())[0] + '.linkinfo'
env.SideEffect( lifile, file )
env.Clean( file, lifile )
return tgt, src
def generate(env):
staticObjBuilder, sharedObjBuilder = SCons.Tool.createObjBuilders(env);
staticObjBuilder.add_action('.cu', '$STATICNVCCCMD')
staticObjBuilder.add_emitter('.cu', CUDANVCCStaticObjectEmitter)
sharedObjBuilder.add_action('.cu', '$SHAREDNVCCCMD')
sharedObjBuilder.add_emitter('.cu', CUDANVCCSharedObjectEmitter)
SCons.Tool.SourceFileScanner.add_scanner('.cu', CUDAScanner)
# default compiler
env['NVCC'] = 'nvcc'
# default flags for the NVCC compiler
env['NVCCFLAGS'] = ''
env['STATICNVCCFLAGS'] = ''
env['SHAREDNVCCFLAGS'] = ''
env['ENABLESHAREDNVCCFLAG'] = '-shared'
# default NVCC commands
env['STATICNVCCCMD'] = '$NVCC -o $TARGET -c $NVCCFLAGS $STATICNVCCFLAGS $SOURCES'
env['SHAREDNVCCCMD'] = '$NVCC -o $TARGET -c $NVCCFLAGS $SHAREDNVCCFLAGS $ENABLESHAREDNVCCFLAG $SOURCES'
# helpers
home=os.environ.get('HOME', '')
programfiles=os.environ.get('PROGRAMFILES', '')
homedrive=os.environ.get('HOMEDRIVE', '')
# find CUDA Toolkit path and set CUDA_TOOLKIT_PATH
try:
cudaToolkitPath = env['CUDA_TOOLKIT_PATH']
except:
paths=[home + '/NVIDIA_CUDA_TOOLKIT',
home + '/Apps/NVIDIA_CUDA_TOOLKIT',
home + '/Apps/NVIDIA_CUDA_TOOLKIT',
home + '/Apps/CudaToolkit',
home + '/Apps/CudaTK',
'/usr/local/NVIDIA_CUDA_TOOLKIT',
'/usr/local/CUDA_TOOLKIT',
'/usr/local/cuda_toolkit',
'/usr/local/CUDA',
'/usr/local/cuda',
'/Developer/NVIDIA CUDA TOOLKIT',
'/Developer/CUDA TOOLKIT',
'/Developer/CUDA',
programfiles + 'NVIDIA Corporation/NVIDIA CUDA TOOLKIT',
programfiles + 'NVIDIA Corporation/NVIDIA CUDA',
programfiles + 'NVIDIA Corporation/CUDA TOOLKIT',
programfiles + 'NVIDIA Corporation/CUDA',
programfiles + 'NVIDIA/NVIDIA CUDA TOOLKIT',
programfiles + 'NVIDIA/NVIDIA CUDA',
programfiles + 'NVIDIA/CUDA TOOLKIT',
programfiles + 'NVIDIA/CUDA',
programfiles + 'CUDA TOOLKIT',
programfiles + 'CUDA',
homedrive + '/CUDA TOOLKIT',
homedrive + '/CUDA']
pathFound = False
for path in paths:
if os.path.isdir(path):
pathFound = True
print 'scons: CUDA Toolkit found in ' + path
cudaToolkitPath = path
break
if not pathFound:
sys.exit("Cannot find the CUDA Toolkit path. Please modify your SConscript or add the path in cudaenv.py")
env['CUDA_TOOLKIT_PATH'] = cudaToolkitPath
# find CUDA SDK path and set CUDA_SDK_PATH
try:
cudaSDKPath = env['CUDA_SDK_PATH']
except:
paths=[home + '/NVIDIA_CUDA_SDK', # i am just guessing here
home + '/Apps/NVIDIA_CUDA_SDK',
home + '/Apps/CudaSDK',
'/usr/local/NVIDIA_CUDA_SDK',
'/usr/local/CUDASDK',
'/usr/local/cuda_sdk',
'/Developer/NVIDIA CUDA SDK',
'/Developer/CUDA SDK',
'/Developer/CUDA',
'/Developer/GPU Computing/C',
programfiles + 'NVIDIA Corporation/NVIDIA CUDA SDK',
programfiles + 'NVIDIA/NVIDIA CUDA SDK',
programfiles + 'NVIDIA CUDA SDK',
programfiles + 'CudaSDK',
homedrive + '/NVIDIA CUDA SDK',
homedrive + '/CUDA SDK',
homedrive + '/CUDA/SDK']
pathFound = False
for path in paths:
if os.path.isdir(path):
pathFound = True
print 'scons: CUDA SDK found in ' + path
cudaSDKPath = path
break
if not pathFound:
sys.exit("Cannot find the CUDA SDK path. Please set env['CUDA_SDK_PATH'] to point to your SDK path")
env['CUDA_SDK_PATH'] = cudaSDKPath
# cuda libraries
if env['PLATFORM'] == 'posix':
cudaSDKSubLibDir = '/linux'
elif env['PLATFORM'] == 'darwin':
cudaSDKSubLibDir = '/darwin'
else:
cudaSDKSubLibDir = ''
# add nvcc to PATH
env.PrependENVPath('PATH', cudaToolkitPath + '/bin')
# add required libraries
env.Append(CPPPATH=[cudaSDKPath + '/common/inc', cudaToolkitPath + '/include'])
env.Append(LIBPATH=[cudaSDKPath + '/lib', cudaSDKPath + '/common/lib' + cudaSDKSubLibDir, cudaToolkitPath + '/lib'])
env.Append(LIBS=['cudart'])
def exists(env):
return env.Detect('nvcc')
The SConscript for building simpleGL:
#!python
env = Environment()
env.Tool('cuda')
env.Append(LIBS=['cutil', 'glut', 'GLEW'])
env.Program('simpleGL', ['simpleGL.cpp', 'simpleGL_kernel.cu'])
The above tool has only been tested on Linux and Mac OS X. If your project doesn't use any of the above paths, you can specify
#!python
env['CUDA_TOOLKIT_PATH'] = path to the CUDA Toolkit
env['CUDA_SDK_PATH'] = path to the CUDA SDK
If the tool finds the paths, it sets the above to the path it found. If you want to include CFLAGS for the nvcc, set the
#!python
env['NVCCFLAGS'] = flags common to static and shared objects
env['STATICNVCCFLAGS'] = flags for static objects
env['SHAREDNVCCFLAGS'] = flags for shared objects
variables. The tool will include automatically the 'cudart' library in LIBS but not cublas nor cufft because you might not need them. Call env.Append as shown in the sample above to add extra libraries after calling env.Tool('cuda').
This is an implementation detail, but i'm posting this too here. I'm not sure if this was the best thing to do, but it seems to work. I added two emitters for .cu files which modify the results of the Scons.Defaults.Static/SharedObjectEmitter to also add .linkinfo suffixed files as a target because the nvcc compiler builds a .o/.obj and a .linkinfo from a .cu file and i needed to somehow tell to SCons to delete the .linkinfo files when scons -c is issued. If you think of a better method for this, please inform me.
After discussions on the usage of the *.linkinfo files plus some googling, I came to the conclusion that the *.linkinfo files shouldn't be returned from the emitters because they were not used in the build chain: NVidia DevTalk on linkinfo. - William Blevins
The implementation has been updated to add *.linkinfo files as side-effects rather than explicit emitter outputs, so that developers using those object builders do not need to filter the emitter returns. Clean targets were added to the side-effect files; thus, resolving the original intention of *.linkinfo files being returned. - William Blevins