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