提交 abd4aff1 authored 作者: Razvan Pascanu's avatar Razvan Pascanu

Merge pull request #3 from lamblin/c_viewop

Fixes after code review
...@@ -70,9 +70,8 @@ class ViewOp(gof.Op): ...@@ -70,9 +70,8 @@ class ViewOp(gof.Op):
"when calling register_deep_copy_op_c_code." % t, "when calling register_deep_copy_op_c_code." % t,
stacklevel=2) stacklevel=2)
return () return ()
version.append((str(t), v)) version.append((str(t), v))
print 'view version:', version
return tuple(version) return tuple(version)
def infer_shape(self, node, input_shapes): def infer_shape(self, node, input_shapes):
...@@ -161,9 +160,8 @@ class DeepCopyOp(gof.Op): ...@@ -161,9 +160,8 @@ class DeepCopyOp(gof.Op):
"when calling register_OutputGuard_c_code." % t, "when calling register_OutputGuard_c_code." % t,
stacklevel=2) stacklevel=2)
return () return ()
version.append((str(t), v)) version.append((str(t), v))
print 'deeop copy version:', version
return tuple(version) return tuple(version)
def c_code(self, node, name, inames, onames, sub): def c_code(self, node, name, inames, onames, sub):
......
...@@ -146,7 +146,7 @@ class SparseType(gof.Type): ...@@ -146,7 +146,7 @@ class SparseType(gof.Type):
def is_valid_value(self, a): def is_valid_value(self, a):
return scipy.sparse.issparse(a) and (a.format == self.format) return scipy.sparse.issparse(a) and (a.format == self.format)
# Register CudaNdarrayType's C code for ViewOp. # Register SparseType's C code for ViewOp.
theano.compile.register_view_op_c_code( theano.compile.register_view_op_c_code(
SparseType, SparseType,
""" """
......
...@@ -56,7 +56,7 @@ class RandomStateType(gof.Type): ...@@ -56,7 +56,7 @@ class RandomStateType(gof.Type):
return False return False
return True return True
# Register CudaNdarrayType's C code for ViewOp. # Register RandomStateType's C code for ViewOp.
theano.compile.register_view_op_c_code( theano.compile.register_view_op_c_code(
RandomStateType, RandomStateType,
""" """
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论