| 177 | } |
| 178 | |
| 179 | daliResult_t daliTensorGetStream( |
| 180 | daliTensor_h tensor, |
| 181 | cudaStream_t *out_stream) { |
| 182 | DALI_PROLOG(); |
| 183 | auto *ptr = ToPointer(tensor); |
| 184 | CHECK_OUTPUT(out_stream); |
| 185 | auto str = ptr->GetStream(); |
| 186 | *out_stream = str.has_value() ? *str : cudaStream_t(-1); |
| 187 | return str.has_value() ? DALI_SUCCESS : DALI_NO_DATA; |
| 188 | DALI_EPILOG(); |
| 189 | } |
| 190 | |
| 191 | daliResult_t daliTensorSetStream( |
| 192 | daliTensor_h tensor, |