提交 fe50c105 authored 作者: Frédéric Bastien's avatar Frédéric Bastien

Merge pull request #3180 from harlouci/props_sandbox_gpuarray

Props sandbox gpuarray
...@@ -323,15 +323,7 @@ gpu_from_host = GpuFromHost() ...@@ -323,15 +323,7 @@ gpu_from_host = GpuFromHost()
class GpuFromCuda(Op): class GpuFromCuda(Op):
view_map = {0: [0]} view_map = {0: [0]}
__props__ = ()
def __eq__(self, other):
return type(self) == type(other)
def __hash__(self):
return hash(type(self))
def __str__(self):
return 'GpuFromCuda'
def make_node(self, x): def make_node(self, x):
from theano.sandbox.cuda import CudaNdarrayType from theano.sandbox.cuda import CudaNdarrayType
...@@ -455,15 +447,7 @@ gpu_from_cuda = GpuFromCuda() ...@@ -455,15 +447,7 @@ gpu_from_cuda = GpuFromCuda()
class CudaFromGpu(Op): class CudaFromGpu(Op):
view_map = {0: [0]} view_map = {0: [0]}
__props__ = ()
def __eq__(self, other):
return type(self) == type(other)
def __hash__(self):
return hash(type(self))
def __str__(self):
return 'CudaFromGpu'
def make_node(self, x): def make_node(self, x):
from theano.sandbox.cuda import CudaNdarrayType from theano.sandbox.cuda import CudaNdarrayType
......
...@@ -185,6 +185,9 @@ class DnnBase(COp): ...@@ -185,6 +185,9 @@ class DnnBase(COp):
class DnnVersion(Op): class DnnVersion(Op):
__props__ = ()
def c_headers(self): def c_headers(self):
return ['cudnn.h'] return ['cudnn.h']
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论