| 365 | } |
| 366 | |
| 367 | void daliSetExternalInputAsync(daliPipelineHandle_t pipe_handle, const char *name, |
| 368 | device_type_t device, const void *data_ptr, |
| 369 | dali_data_type_t data_type, const int64_t *shapes, |
| 370 | int sample_dim, const char *layout_str, cudaStream_t stream, |
| 371 | unsigned int flags) { |
| 372 | switch (device) { |
| 373 | case device_type_t::CPU: |
| 374 | SetExternalInput<CPUBackend>(pipe_handle, name, data_ptr, data_type, shapes, sample_dim, |
| 375 | layout_str, stream, flags); |
| 376 | return; |
| 377 | case device_type_t::GPU: |
| 378 | SetExternalInput<GPUBackend>(pipe_handle, name, data_ptr, data_type, shapes, sample_dim, |
| 379 | layout_str, stream, flags); |
| 380 | return; |
| 381 | default: |
| 382 | DALI_FAIL(dali::make_string("Unknown device: ", device)); |
| 383 | } |
| 384 | } |
| 385 | |
| 386 | |
| 387 | void daliSetExternalInputTensors(daliPipelineHandle_t pipe_handle, const char *name, |