提交 25d6298f authored 作者: james@x40.unstable's avatar james@x40.unstable

BUGFIX: gpu_conv - added kshp to hash and eq

上级 12934a1f
...@@ -296,7 +296,8 @@ class GpuConv(Op): ...@@ -296,7 +296,8 @@ class GpuConv(Op):
and self.logical_kern_hw == other.logical_kern_hw \ and self.logical_kern_hw == other.logical_kern_hw \
and self.logical_kern_align_top == other.logical_kern_align_top \ and self.logical_kern_align_top == other.logical_kern_align_top \
and self.version == other.version \ and self.version == other.version \
and self.verbose == other.verbose and self.verbose == other.verbose \
and self.kshp == other.kshp
def __hash__(self): def __hash__(self):
# don't use hash(self.version) as hash(-1)==-2 and hash(-2)==-2 in python! # don't use hash(self.version) as hash(-1)==-2 and hash(-2)==-2 in python!
...@@ -307,7 +308,8 @@ class GpuConv(Op): ...@@ -307,7 +308,8 @@ class GpuConv(Op):
^ hash(self.logical_kern_hw) \ ^ hash(self.logical_kern_hw) \
^ hash(self.logical_kern_align_top) \ ^ hash(self.logical_kern_align_top) \
^ self.version \ ^ self.version \
^ self.verbose ^ self.verbose \
^ self.kshp
def __str__(self): def __str__(self):
return '%s{%s, %s, %s, %s, %s}' %(self.__class__.__name__, return '%s{%s, %s, %s, %s, %s}' %(self.__class__.__name__,
...@@ -336,7 +338,7 @@ class GpuConv(Op): ...@@ -336,7 +338,7 @@ class GpuConv(Op):
return ['cuda_ndarray.cuh','<stdio.h>'] return ['cuda_ndarray.cuh','<stdio.h>']
def c_code_cache_version(self): def c_code_cache_version(self):
return (0,6) return (0,7)
def c_support_code_apply(self, node, nodename): def c_support_code_apply(self, node, nodename):
return open(os.path.join(os.path.split(__file__)[0],'conv_kernel.cu')).read()+\ return open(os.path.join(os.path.split(__file__)[0],'conv_kernel.cu')).read()+\
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论