提交 cb89010f authored 作者: James Bergstra's avatar James Bergstra

moved the code that creates the libcuda_ndarray symlink, in hopes of fixing buildbot

上级 afa753c4
...@@ -131,6 +131,12 @@ try: ...@@ -131,6 +131,12 @@ try:
code, code,
location=cuda_ndarray_loc, location=cuda_ndarray_loc,
include_dirs=[cuda_path], libs=['cublas']) include_dirs=[cuda_path], libs=['cublas'])
from cuda_ndarray.cuda_ndarray import *
except Exception, e:
error( "Failed to compile cuda_ndarray.cu: %s" % str(e))
set_cuda_disabled()
if cuda_available:
# If necessary, # If necessary,
# create a symlink called libcuda_ndarray.so # create a symlink called libcuda_ndarray.so
# which nvcc_module_compile_str uses when linking # which nvcc_module_compile_str uses when linking
...@@ -139,12 +145,7 @@ try: ...@@ -139,12 +145,7 @@ try:
open(libcuda_ndarray_so).close() open(libcuda_ndarray_so).close()
except IOError: except IOError:
os.symlink(cuda_ndarray_so, libcuda_ndarray_so) os.symlink(cuda_ndarray_so, libcuda_ndarray_so)
from cuda_ndarray.cuda_ndarray import *
except Exception, e:
error( "Failed to compile cuda_ndarray.cu: %s" % str(e))
set_cuda_disabled()
if cuda_available:
try: try:
gpu_init() gpu_init()
cuda_available = True cuda_available = True
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论