提交 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()
class GpuFromCuda(Op):
view_map = {0: [0]}
def __eq__(self, other):
return type(self) == type(other)
......@@ -334,6 +336,8 @@ gpu_from_cuda = GpuFromCuda()
class CudaFromGpu(Op):
view_map = {0: [0]}
def __eq__(self, other):
return type(self) == type(other)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论