提交 85023583 authored 作者: Pierre Luc Carrier's avatar Pierre Luc Carrier

Added methods c_headers() and c_compiler() to GpuImages2Neibs

上级 f32dff50
...@@ -2,6 +2,7 @@ ...@@ -2,6 +2,7 @@
from theano import Op, Apply from theano import Op, Apply
from theano.gof import local_optimizer from theano.gof import local_optimizer
from theano.sandbox.cuda import cuda_available from theano.sandbox.cuda import cuda_available
from theano.sandbox.cuda.nvcc_compiler import NVCC_compiler
from theano.sandbox.neighbours import Images2Neibs from theano.sandbox.neighbours import Images2Neibs
...@@ -35,6 +36,13 @@ class GpuImages2Neibs(Images2Neibs, Op): ...@@ -35,6 +36,13 @@ class GpuImages2Neibs(Images2Neibs, Op):
def c_code_cache_version(self): def c_code_cache_version(self):
return (8,) return (8,)
def c_headers(self):
return ['cuda.h', '<compyte/extension.h>', '<numpy_compat.h>',
'<compyte/ext_cuda.h>']
def c_compiler(self):
return NVCC_compiler
def c_support_code_apply(self, node, nodename): def c_support_code_apply(self, node, nodename):
mode = self.mode mode = self.mode
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论