提交 88fa607c authored 作者: Frederic's avatar Frederic

Always convert broadcast to bool in CudaNdarrayType as done in TensorType and GpuArrayType

上级 39da3f0c
...@@ -81,7 +81,7 @@ class CudaNdarrayType(Type): ...@@ -81,7 +81,7 @@ class CudaNdarrayType(Type):
raise TypeError('%s only supports dtype float32 for now. Tried ' raise TypeError('%s only supports dtype float32 for now. Tried '
'using dtype %s for variable %s' % 'using dtype %s for variable %s' %
(self.__class__.__name__, dtype, name)) (self.__class__.__name__, dtype, name))
self.broadcastable = tuple(broadcastable) self.broadcastable = tuple(bool(b) for b in broadcastable)
self.name = name self.name = name
self.dtype_specs() # error checking is done there self.dtype_specs() # error checking is done there
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论