提交 fa143f13 authored 作者: Chiheb Trabelsi's avatar Chiheb Trabelsi

type.py has been modified in order to respect the flake8 style.

上级 4fd12c42
...@@ -259,8 +259,8 @@ class CudaNdarrayType(Type): ...@@ -259,8 +259,8 @@ class CudaNdarrayType(Type):
'complex64': (complex, 'theano_complex64', 'complex64': (complex, 'theano_complex64',
'NPY_COMPLEX64')}[self.dtype] 'NPY_COMPLEX64')}[self.dtype]
except KeyError: except KeyError:
raise TypeError("Unsupported dtype for %s: %s" % ( raise TypeError("Unsupported dtype for %s: %s" %
self.__class__.__name__, self.dtype)) (self.__class__.__name__, self.dtype))
def __eq__(self, other): def __eq__(self, other):
""" """
...@@ -271,9 +271,10 @@ class CudaNdarrayType(Type): ...@@ -271,9 +271,10 @@ class CudaNdarrayType(Type):
other.broadcastable == self.broadcastable) other.broadcastable == self.broadcastable)
def convert_variable(self, var): def convert_variable(self, var):
if (type(self) == type(var.type) and if (isinstance(self, type(var.type)) and
self.ndim == var.type.ndim and self.ndim == var.type.ndim and
all(sb == ob or ob for sb, ob in zip(self.broadcastable, all(sb == ob or ob for sb, ob in zip(
self.broadcastable,
var.type.broadcastable))): var.type.broadcastable))):
return theano.tensor.patternbroadcast(var, self.broadcastable) return theano.tensor.patternbroadcast(var, self.broadcastable)
...@@ -312,7 +313,7 @@ class CudaNdarrayType(Type): ...@@ -312,7 +313,7 @@ class CudaNdarrayType(Type):
return self.name return self.name
else: else:
b = self.broadcastable b = self.broadcastable
#bcast = str(self.broadcastable) # bcast = str(self.broadcastable)
if not numpy.any(b): if not numpy.any(b):
s = "%iD" % len(b) s = "%iD" % len(b)
else: else:
...@@ -327,7 +328,7 @@ class CudaNdarrayType(Type): ...@@ -327,7 +328,7 @@ class CudaNdarrayType(Type):
def __repr__(self): def __repr__(self):
return str(self) return str(self)
#"CudaNdarrayType{%s, %s}" % (str(self.dtype), str(self.broadcastable)) # "CudaNdarrayType{%s, %s}" % (str(self.dtype), str(self.broadcastable))
def c_declare(self, name, sub, check_input=True): def c_declare(self, name, sub, check_input=True):
return """ CudaNdarray * %(name)s;""" % locals() return """ CudaNdarray * %(name)s;""" % locals()
...@@ -563,8 +564,7 @@ theano.compile.register_deep_copy_op_c_code( ...@@ -563,8 +564,7 @@ theano.compile.register_deep_copy_op_c_code(
CudaNdarray_HOST_DIMS(%(oname)s)[i]) { CudaNdarray_HOST_DIMS(%(oname)s)[i]) {
alloc = true; alloc = true;
break; break;
} }}
}
if(alloc) { if(alloc) {
Py_XDECREF(%(oname)s); Py_XDECREF(%(oname)s);
%(oname)s = (CudaNdarray*)CudaNdarray_Copy(%(iname)s); %(oname)s = (CudaNdarray*)CudaNdarray_Copy(%(iname)s);
...@@ -581,8 +581,7 @@ theano.compile.register_deep_copy_op_c_code( ...@@ -581,8 +581,7 @@ theano.compile.register_deep_copy_op_c_code(
%(fail)s; %(fail)s;
} }
} }
""", """, version=3)
version=3)
# THIS WORKS But CudaNdarray instances don't compare equal to one # THIS WORKS But CudaNdarray instances don't compare equal to one
...@@ -608,5 +607,5 @@ def CudaNdarray_pickler(cnda): ...@@ -608,5 +607,5 @@ def CudaNdarray_pickler(cnda):
# In case cuda is not imported. # In case cuda is not imported.
if cuda is not None: if cuda is not None:
copyreg.pickle(cuda.CudaNdarray, CudaNdarray_pickler, copyreg.pickle(
CudaNdarray_unpickler) cuda.CudaNdarray, CudaNdarray_pickler, CudaNdarray_unpickler)
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论