提交 06f3cd73 authored 作者: f0k's avatar f0k

Use __props__ and drop grad() in GpuAlloc*

上级 ad8571e0
...@@ -3259,14 +3259,7 @@ class GpuSplit(tensor.Split, GpuOp): ...@@ -3259,14 +3259,7 @@ class GpuSplit(tensor.Split, GpuOp):
class GpuAllocEmpty(GpuOp): class GpuAllocEmpty(GpuOp):
"""Implement Alloc on the gpu, but without initializing memory.""" """Implement Alloc on the gpu, but without initializing memory."""
def __eq__(self, other): __props__ = ()
return type(self) == type(other)
def __hash__(self):
return hash(type(self))
def __str__(self):
return self.__class__.__name__
def make_node(self, *shape): def make_node(self, *shape):
sh = [tensor.as_tensor_variable(s) for s in shape] sh = [tensor.as_tensor_variable(s) for s in shape]
...@@ -3325,10 +3318,6 @@ class GpuAllocEmpty(GpuOp): ...@@ -3325,10 +3318,6 @@ class GpuAllocEmpty(GpuOp):
def infer_shape(self, node, input_shapes): def infer_shape(self, node, input_shapes):
return [node.inputs] return [node.inputs]
def grad(self, inputs, grads):
gout, = grads
return [None for i in inputs]
def c_code_cache_version(self): def c_code_cache_version(self):
return (1,) return (1,)
...@@ -3346,15 +3335,11 @@ class GpuAlloc(GpuAllocEmpty): ...@@ -3346,15 +3335,11 @@ class GpuAlloc(GpuAllocEmpty):
cudaMemset that is faster. cudaMemset that is faster.
""" """
__props__ = ('memset_0',)
def __init__(self, memset_0=False): def __init__(self, memset_0=False):
self.memset_0 = memset_0 self.memset_0 = memset_0
def __eq__(self, other):
return type(self) == type(other) and self.memset_0 == other.memset_0
def __hash__(self):
return hash(type(self)) ^ hash(self.memset_0)
def __str__(self): def __str__(self):
#Hide the memset parameter when not used to prevent confusion. #Hide the memset parameter when not used to prevent confusion.
if self.memset_0: if self.memset_0:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论