Function AccessOrderFromPythonStreamObj

dali/python/backend_impl.cc:647–670

}

AccessOrder AccessOrderFromPythonStreamObj(const py::object &cuda_stream) {
  AccessOrder order;
  if (!cuda_stream.is_none()) {
    auto cuda_stream_interface = getattr(cuda_stream, "__cuda_stream__", py::none());
    if (!cuda_stream_interface.is_none()) {
      auto [version, stream_ptr] = cuda_stream_interface().cast<std::tuple<int, uintptr_t>>();
      cudaStream_t stream = reinterpret_cast<cudaStream_t>(stream_ptr);
      order = AccessOrder(stream);
    } else if (py::isinstance<py::int_>(cuda_stream)) {
      cudaStream_t stream = reinterpret_cast<cudaStream_t>(py::cast<uintptr_t>(cuda_stream));
      order = AccessOrder(stream);
    } else if (py::hasattr(cuda_stream, "value")) {
      cudaStream_t stream = static_cast<cudaStream_t>(ctypes_void_ptr(cuda_stream));
      order = AccessOrder(stream);
    } else if (auto cuda_stream_prop = getattr(cuda_stream, "cuda_stream", py::none())) {
      return AccessOrderFromPythonStreamObj(cuda_stream_prop);
    } else if (auto handle_prop = getattr(cuda_stream, "handle", py::none())) {
      return AccessOrderFromPythonStreamObj(handle_prop);
    }
  } else {
    order = AccessOrder::host();
  }
  return order;
}

/**
 * Pipeline output descriptor.