提交 75a4738b authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

No need to ask for NVCC since we have a CudaNdarrayType parameter which will do it for us.

上级 a2d22152
......@@ -257,10 +257,6 @@ class GpuFromCuda(Op):
def c_libraries(self):
return ['cudart', 'cublas', 'cuda']
def c_compiler(self):
from theano.sandbox.cuda.nvcc_compiler import NVCC_compiler
return NVCC_compiler
def c_support_code(self):
return """
CUcontext (*cuda_get_ctx)(void *ctx);
......@@ -405,10 +401,6 @@ class CudaFromGpu(Op):
def c_libraries(self):
return ['cudart', 'cublas', 'cuda']
def c_compiler(self):
from theano.sandbox.cuda.nvcc_compiler import NVCC_compiler
return NVCC_compiler
def c_support_code(self):
return """
CUcontext (*cuda_get_ctx)(void *ctx);
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论