MCPcopy Create free account
hub / github.com/NVIDIA/DALI / GetBufferPlacement

Method GetBufferPlacement

dali/c_api_2/data_objects.h:298–305  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

296 }
297
298 daliBufferPlacement_t GetBufferPlacement() const override {
299 daliBufferPlacement_t placement;
300 placement.device_id = t_->device_id();
301 StorageDevice dev = backend_to_storage_device<Backend>::value;
302 placement.device_type = static_cast<daliStorageDevice_t>(dev);
303 placement.pinned = t_->is_pinned();
304 return placement;
305 }
306
307 void SetStream(std::optional<cudaStream_t> stream, bool synchronize) override {
308 t_->set_order(stream.has_value() ? AccessOrder(*stream) : AccessOrder::host(), synchronize);

Callers 3

FeedInputMethod · 0.45

Calls 1

device_idMethod · 0.45

Tested by

no test coverage detected