提交 e5f73eb4 authored 作者: Aleksandar Botev's avatar Aleksandar Botev

Small change

上级 1f1e0369
...@@ -31,6 +31,9 @@ class GpuImages2Neibs(GpuKernelBase, Images2Neibs, Op): ...@@ -31,6 +31,9 @@ class GpuImages2Neibs(GpuKernelBase, Images2Neibs, Op):
self.mode = mode self.mode = mode
def make_node(self, ten4, neib_shape, neib_step=None): def make_node(self, ten4, neib_shape, neib_step=None):
# TODO: I don't know why, but without this the tests fail?
neib_shape.eval()
####
ten4 = as_gpuarray_variable(ten4, infer_context_name(ten4)) ten4 = as_gpuarray_variable(ten4, infer_context_name(ten4))
neib_shape = T.as_tensor_variable(neib_shape) neib_shape = T.as_tensor_variable(neib_shape)
if neib_step is None: if neib_step is None:
...@@ -53,7 +56,7 @@ class GpuImages2Neibs(GpuKernelBase, Images2Neibs, Op): ...@@ -53,7 +56,7 @@ class GpuImages2Neibs(GpuKernelBase, Images2Neibs, Op):
return node.inputs[0].type.context return node.inputs[0].type.context
def c_code_cache_version(self): def c_code_cache_version(self):
return (11,) return (13,)
def c_headers(self): def c_headers(self):
return ['<numpy_compat.h>', '<gpuarray/types.h>'] return ['<numpy_compat.h>', '<gpuarray/types.h>']
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论