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)