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

Method GetStream

dali/c_api_2/data_objects.h:326–332  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

324 }
325
326 std::optional<cudaStream_t> GetStream() const override {
327 auto o = t_->order();
328 if (o.is_device())
329 return o.stream();
330 else
331 return std::nullopt;
332 }
333
334 std::optional<cudaEvent_t> GetReadyEvent() const override {
335 auto &e = t_->ready_event();

Callers 9

ExposeTensorFunction · 0.45
ExposeTensorListCPUFunction · 0.45
ExposeTesorListGPUFunction · 0.45
ExposePipelineFunction · 0.45
daliTensorGetStreamFunction · 0.45
daliTensorListGetStreamFunction · 0.45

Calls 1

streamMethod · 0.45

Tested by

no test coverage detected