提交 31ddaabe authored 作者: Frédéric Bastien's avatar Frédéric Bastien

Merge pull request #1695 from abergeron/fix_npcompat

Fix include path for numpy_compat.h following change in compyte.
...@@ -563,7 +563,7 @@ class GpuAlloc(HideC, Alloc): ...@@ -563,7 +563,7 @@ class GpuAlloc(HideC, Alloc):
return Apply(self, [value] + res.inputs[1:], [otype()]) return Apply(self, [value] + res.inputs[1:], [otype()])
def c_headers(self): def c_headers(self):
return ['<compyte/numpy_compat.h>'] return ['<numpy_compat.h>']
def perform(self, node, inputs, outs): def perform(self, node, inputs, outs):
out, = outs out, = outs
......
...@@ -191,7 +191,7 @@ class GpuDot22(BlasOp, Dot22): ...@@ -191,7 +191,7 @@ class GpuDot22(BlasOp, Dot22):
def c_headers(self): def c_headers(self):
ret = super(GpuDot22, self).c_headers() ret = super(GpuDot22, self).c_headers()
return ret + ['<compyte/numpy_compat.h>'] return ret + ['<numpy_compat.h>']
gpu_dot22 = GpuDot22() gpu_dot22 = GpuDot22()
......
...@@ -183,7 +183,7 @@ class GpuConv(gof.Op): ...@@ -183,7 +183,7 @@ class GpuConv(gof.Op):
def c_headers(self): def c_headers(self):
return ['<stdio.h>', 'cuda.h', return ['<stdio.h>', 'cuda.h',
'<compyte/extension.h>', '<compyte/numpy_compat.h>'] '<compyte/extension.h>', '<numpy_compat.h>']
def c_code_cache_version(self): def c_code_cache_version(self):
# raise this whenever modifying any of the support_code_files # raise this whenever modifying any of the support_code_files
......
...@@ -165,7 +165,7 @@ class GpuElemwise(HideC, Elemwise): ...@@ -165,7 +165,7 @@ class GpuElemwise(HideC, Elemwise):
return ElemwiseKernel(None, inps+outs, kop, preamble=support_code) return ElemwiseKernel(None, inps+outs, kop, preamble=support_code)
def c_headers(self): def c_headers(self):
return ['cuda.h', '<compyte/extension.h>', '<compyte/numpy_compat.h>'] return ['cuda.h', '<compyte/extension.h>', '<numpy_compat.h>']
def c_compiler(self): def c_compiler(self):
return NVCC_compiler return NVCC_compiler
......
...@@ -44,7 +44,7 @@ class GpuCrossentropySoftmaxArgmax1HotWithBias(Op): ...@@ -44,7 +44,7 @@ class GpuCrossentropySoftmaxArgmax1HotWithBias(Op):
return Apply(self, [x, b, y_idx], [nll, sm, am]) return Apply(self, [x, b, y_idx], [nll, sm, am])
def c_headers(self): def c_headers(self):
return ['cuda.h', '<compyte/extension.h>', '<compyte/numpy_compat.h>'] return ['cuda.h', '<compyte/extension.h>', '<numpy_compat.h>']
def c_support_code_apply(self, node, nodename): def c_support_code_apply(self, node, nodename):
dtype_x = node.inputs[0].dtype dtype_x = node.inputs[0].dtype
...@@ -297,7 +297,7 @@ class GpuCrossentropySoftmax1HotWithBiasDx(Op): ...@@ -297,7 +297,7 @@ class GpuCrossentropySoftmax1HotWithBiasDx(Op):
return (6,) return (6,)
def c_headers(self): def c_headers(self):
return ['cuda.h', '<compyte/extension.h>', '<compyte/numpy_compat.h>'] return ['cuda.h', '<compyte/extension.h>', '<numpy_compat.h>']
def c_compiler(self): def c_compiler(self):
return NVCC_compiler return NVCC_compiler
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论