提交 04dbda06 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Implement get_shape_info and get_size.

上级 57e6678e
...@@ -126,6 +126,15 @@ class GpuArrayType(Type): ...@@ -126,6 +126,15 @@ class GpuArrayType(Type):
def __str__(self): def __str__(self):
return "GpuArray<%s>" % (self.dtype,) return "GpuArray<%s>" % (self.dtype,)
def get_shape_info(self, obj):
return obj.shape
def get_size(self, shape_info):
if shape_info:
return numpy.prod(shape_info) * numpy.dtype(self.dtype).itemsize
else:
return numpy.dtype(self.dtype).itemsize
def c_declare(self, name, sub): def c_declare(self, name, sub):
return "GpuArrayObject *%s;" % (name,) return "GpuArrayObject *%s;" % (name,)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论