Function holoscan::inference::allocate_buffers

Holoscan v2.2.0
InferStatus holoscan::inference::allocate_buffers(DataMap &buffers, std::vector<int64_t> &dims, holoinfer_datatype datatype, const std::string &keyname, bool allocate_cuda, int device_id)

Allocate buffer on host and device.

  • buffersMap with keyword as model name or tensor name, and value as DataBuffer. The map is populated in this function.

  • dims – Dimension of the allocation

  • datatype – Data type of the buffer

  • keyname – Storage name in the map against the created DataBuffer

  • allocate_cuda – flag to allocate cuda buffer

  • device_id – GPU ID to allocate buffers on


InferStatus with appropriate code and message

Previous Function holoscan::inference::__attribute__
Next Function holoscan::inference::build_engine
© Copyright 2022-2024, NVIDIA. Last updated on Jul 17, 2024.