Method from_dlpack

imperative/python/src/dlpack_convertor.cpp:246–270

}

ValueRef mgb::imperative::from_dlpack(DLManagedTensor* dlMTensor, int stream = 0) {
    std::function<void(void*)> deleter_dispatch = [dlMTensor](void*) {
        if (dlMTensor->deleter) {
            dlMTensor->deleter(dlMTensor);
        }
    };

    DType tensor_type = get_tensor_type(dlMTensor->dl_tensor.dtype);
    CompNode tensor_device = get_tensor_device(dlMTensor->dl_tensor.device, stream);
    DeviceTensorStorage storage;
    size_t dtype_size = tensor_type.size();
    size_t ndim = dlMTensor->dl_tensor.ndim;
    TensorShape tensor_shape = ptr2shape(dlMTensor->dl_tensor.shape, ndim);

    storage.reset(
            tensor_device, tensor_shape.total_nr_elems() * dtype_size,
            {static_cast<dt_byte*>(dlMTensor->dl_tensor.data), deleter_dispatch});

    ValueShape shapevalue = ValueShape::from(tensor_shape);
    ValueRef val = imperative::apply(
            CreateTensor(
                    CreateTensor::Common, tensor_device, tensor_type, shapevalue, {}),
            DeviceStorage::make(storage))[0];
    return val;
};