提交 1bedef60 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Add view_map notifications for GpuFromCuda and CudaFromGpu

上级 24d1bb67
...@@ -196,6 +196,8 @@ gpu_from_host = GpuFromHost() ...@@ -196,6 +196,8 @@ gpu_from_host = GpuFromHost()
class GpuFromCuda(Op): class GpuFromCuda(Op):
view_map = {0: [0]}
def __eq__(self, other): def __eq__(self, other):
return type(self) == type(other) return type(self) == type(other)
...@@ -334,6 +336,8 @@ gpu_from_cuda = GpuFromCuda() ...@@ -334,6 +336,8 @@ gpu_from_cuda = GpuFromCuda()
class CudaFromGpu(Op): class CudaFromGpu(Op):
view_map = {0: [0]}
def __eq__(self, other): def __eq__(self, other):
return type(self) == type(other) return type(self) == type(other)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论