def to_gputensor(a):
assert a.flags.c_contiguous or a.flags.f_contiguous
return GPUTensor(a.shape,
dtype=a.dtype,
base=a,
gpudata=a.gpudata + a.offset,
strides=a.strides,
is_trans=a.flags.f_contiguous)
def to_gputensor(a):
assert a.is_c_contiguous
strides = tuple(s * 4 for s in a.strides)
return GPUTensor(a.shape,
dtype=np.dtype(a.dtype),
base=a,
gpudata=a.gpudata,
strides=strides,
is_trans=False)