Function SetExternalInput

dali/c_api/c_api.cc:121–158

template <typename Backend>
void SetExternalInput(daliPipelineHandle_t pipe_handle, const char *name, const void *data_ptr,
                      dali_data_type_t data_type, const int64_t *shapes, int sample_dim,
                      const char *layout_str, cudaStream_t stream = 0, unsigned int flags = 0) {
  dali::Pipeline *pipeline = (*pipe_handle)->pipeline.get();
  auto *bs_map = &(*pipe_handle)->batch_size_map;
  auto *data_id_map = &(*pipe_handle)->data_id_map;
  auto curr_batch_size = PopCurrBatchSize(bs_map, pipeline->max_batch_size(), name);
  std::vector<int64_t> shapes_tmp(shapes, shapes + sample_dim * curr_batch_size);
  dali::TensorListShape<> tl_shape(std::move(shapes_tmp), curr_batch_size, sample_dim);
  dali::TensorLayout layout{};
  if (layout_str != nullptr) {
    layout = dali::TensorLayout(layout_str);
  }
  dali::TensorList<Backend> data;
  auto type_id = static_cast<dali::DALIDataType>(data_type);
  auto elem_sizeof = dali::TypeTable::GetTypeInfo(type_id).size();
  // We cast away the const from data_ptr, as there is no other way of passing it to the
  // TensorList, as we must also set the shape and type metadata.
  // It is passed further as const TensorList, so it's data cannot be modified.
  AccessOrder order;
  if (std::is_same_v<Backend, GPUBackend> || (flags & DALI_ext_pinned))
    order = AccessOrder(stream);
  else
    order = AccessOrder::host();
  // We do not support feeding memory cross-device, it is assumed it's on the current device
  // that is tied to the pipeline.
  int device_id = pipeline->device_id();
  data.ShareData(std::shared_ptr<void>(const_cast<void *>(data_ptr), [](void *) {}),
                 tl_shape.num_elements() * elem_sizeof, flags & DALI_ext_pinned, tl_shape, type_id,
                 device_id, order);
  data.SetLayout(layout);

  auto data_id = data_id_map->extract(name);

  pipeline->SetExternalInput(name, data, order, flags & DALI_ext_force_sync,
                             flags & DALI_use_copy_kernel, GetExternalSourceCopyMode(flags),
                             data_id ? std::make_optional(data_id.mapped()) : std::nullopt);
}


template<typename Backend>