some small fix to make it work when the end don't have CUDA_ROOT, but that the…

nvcc_compiler.py

@@ -34,10 +34,12 @@ def nvcc_module_compile_str(module_name, src_code, location=None, include_dirs=[
     preargs= [] if preargs is None else list(preargs)
     preargs.append('-fPIC')
     no_opt = False
-
+    cuda_root = os.getenv('CUDA_ROOT')
     include_dirs = std_include_dirs() + include_dirs
     libs = std_libs() + ['cudart'] + libs
-    lib_dirs = std_lib_dirs() + [os.path.join(os.getenv('CUDA_ROOT'), 'lib')] + lib_dirs
+    lib_dirs = std_lib_dirs() + lib_dirs
+    if cuda_root:
+        lib_dirs.append(os.path.join(cuda_root, 'lib'))
 
     cppfilename = os.path.join(location, 'mod.cu')
     cppfile = file(cppfilename, 'w')

type.py

@@ -235,12 +235,18 @@ class CudaNdarrayType(Type):
 
     def c_header_dirs(self):
         """Override `CLinkerOp.c_headers` """
-        return [os.path.dirname(cuda_ndarray.__file__),
-                os.path.join(os.getenv("CUDA_ROOT"),'include')]
+        ret = [os.path.dirname(cuda_ndarray.__file__)]
+        cuda_root = os.getenv("CUDA_ROOT")
+        if cuda_root:
+            ret.append(os.path.join(cuda_root,'include'))
+        return ret
 
     def c_lib_dirs(self):
-        return [os.path.dirname(cuda_ndarray.__file__),
-                os.path.join(os.getenv("CUDA_ROOT"),'lib')]
+        ret = [os.path.dirname(cuda_ndarray.__file__)]
+        cuda_root = os.getenv("CUDA_ROOT")
+        if cuda_root:
+            ret.append(os.path.join(cuda_root,'lib'))
+        return ret
 
     def c_libraries(self):
         return ['cuda_ndarray', 'cudart']