Method AttachBuffer

dali/c_api_2/data_objects.h:247–296

  }

  void AttachBuffer(
      int ndim,
      const int64_t *shape,
      daliDataType_t dtype,
      const char *layout,
      void *data,
      daliDeleter_t deleter) override {
    ValidateShape(ndim, shape);
    Validate(dtype);
    if (layout)
      Validate(TensorLayout(layout), ndim);

    TensorShape<> tshape(make_cspan(shape, ndim));
    size_t num_elements = volume(tshape);
    if (num_elements > 0 && !data)
      throw std::invalid_argument("The data buffer must not be NULL for a non-empty tensor.");

    TensorLayout new_layout = {};

    if (!layout) {
      if (ndim == t_->ndim())
        new_layout = t_->GetLayout();
    } else {
      new_layout = layout;
      Validate(new_layout, ndim);
    }

    t_->Reset();
    auto type_info = TypeTable::GetTypeInfo(dtype);
    auto element_size = type_info.size();

    std::shared_ptr<void> buffer;
    if (!deleter.delete_buffer && !deleter.destroy_context) {
      buffer = std::shared_ptr<void>(data, [](void *){});
    } else {
      buffer = std::shared_ptr<void>(data, BufferDeleter{deleter, t_->order()});
    }

    t_->ShareData(
      std::move(buffer),
      num_elements * element_size,
      t_->is_pinned(),
      tshape,
      dtype,
      t_->device_id(),
      t_->order());

    if (layout)
      t_->SetLayout(new_layout);
  }

  daliBufferPlacement_t GetBufferPlacement() const override {
    daliBufferPlacement_t placement;