worker_builder.h

 
#pragma once
 
#include <memory>
#include <utility>
 
#include <ucxx/buffer.h>
 
namespace ucxx {
 
// Forward declarations
class Context;
class Worker;
 
class WorkerBuilder final {
 public:
  explicit WorkerBuilder(std::shared_ptr<Context> context);
 
  ~WorkerBuilder();
 
  WorkerBuilder(const WorkerBuilder& other);
  WorkerBuilder& operator=(const WorkerBuilder& other);
  WorkerBuilder(WorkerBuilder&&) noexcept;
  WorkerBuilder& operator=(WorkerBuilder&&) noexcept;
 
  operator std::shared_ptr<Worker>();
 
  WorkerBuilder& delayedSubmission(bool enable = true);
 
  WorkerBuilder& pythonFuture(bool enable = true);
 
  WorkerBuilder& requestAttributes(bool enable = true);
 
  WorkerBuilder& cudaBufferType(BufferType bufferType);
 
  [[nodiscard]] std::shared_ptr<Worker> build();
 
 private:
  struct Impl;
  std::unique_ptr<Impl> _impl;
};
 
[[nodiscard]] inline WorkerBuilder workerBuilder(std::shared_ptr<Context> context)
{
  return WorkerBuilder(std::move(context));
}
 
}  // namespace ucxx