mirror of
https://github.com/wassname/ray.git
synced 2026-07-17 11:32:33 +08:00
42 lines
1.0 KiB
C++
42 lines
1.0 KiB
C++
|
|
#pragma once
|
|
|
|
#include "ray/core.h"
|
|
|
|
namespace ray {
|
|
namespace api {
|
|
|
|
template <typename ReturnType>
|
|
class TaskCaller {
|
|
public:
|
|
TaskCaller();
|
|
|
|
TaskCaller(RayRuntime *runtime, RemoteFunctionPtrHolder ptr,
|
|
std::vector<std::unique_ptr<::ray::TaskArg>> &&args);
|
|
|
|
ObjectRef<ReturnType> Remote();
|
|
|
|
private:
|
|
RayRuntime *runtime_;
|
|
RemoteFunctionPtrHolder ptr_;
|
|
std::vector<std::unique_ptr<::ray::TaskArg>> args_;
|
|
};
|
|
|
|
// ---------- implementation ----------
|
|
|
|
template <typename ReturnType>
|
|
TaskCaller<ReturnType>::TaskCaller() {}
|
|
|
|
template <typename ReturnType>
|
|
TaskCaller<ReturnType>::TaskCaller(RayRuntime *runtime, RemoteFunctionPtrHolder ptr,
|
|
std::vector<std::unique_ptr<::ray::TaskArg>> &&args)
|
|
: runtime_(runtime), ptr_(ptr), args_(std::move(args)) {}
|
|
|
|
template <typename ReturnType>
|
|
ObjectRef<ReturnType> TaskCaller<ReturnType>::Remote() {
|
|
auto returned_object_id = runtime_->Call(ptr_, args_);
|
|
return ObjectRef<ReturnType>(returned_object_id);
|
|
}
|
|
} // namespace api
|
|
} // namespace ray
|