// Copyright (c) 2022 PaddlePaddle Authors. All Rights Reserved. // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. // You may obtain a copy of the License at // // http://www.apache.org/licenses/LICENSE-2.0 // // Unless required by applicable law or agreed to in writing, software // distributed under the License is distributed on an "AS IS" BASIS, // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // See the License for the specific language governing permissions and // limitations under the License. #pragma once #include #include #include #include "paddle/fluid/framework/executor.h" #include "paddle/fluid/framework/program_desc.h" #include "paddle/fluid/framework/scope.h" #include "paddle/fluid/framework/variable.h" #include "paddle/fluid/jit/base_function.h" #include "paddle/fluid/jit/function_schema.h" #include "paddle/fluid/jit/function_utils.h" namespace paddle { namespace jit { class ExecutorFunction : public BaseFunction { public: ExecutorFunction(const std::shared_ptr &info, const Name2VariableMap ¶ms_dict, const phi::Place &place) : info_(info), place_(place), inner_exe_(place_) { utils::ShareParamsIntoScope(info_->ParamNames(), params_dict, &scope_); VLOG(6) << framework::GenScopeTreeDebugInfo(&scope_); info_->RemoveDescFeedFetch(); } ~ExecutorFunction() noexcept {} std::vector operator()(const std::vector &inputs) { auto dense_tensors = utils::ToDenseTensors(inputs); return utils::ToTensors(this->operator()(dense_tensors)); } std::vector operator()(const std::vector &inputs) { utils::ShareIntoScope(info_->InputArgNames(), inputs, &scope_); inner_exe_.Run(info_->ProgramDesc(), &scope_, /*blockID=*/0, false, true, info_->OutputArgNames()); std::vector res; utils::FetchOuts(info_->OutputArgNames(), scope_, &res); return res; } const std::shared_ptr &Info() const { return info_; } private: std::shared_ptr info_; framework::Scope scope_; phi::Place place_; framework::Executor inner_exe_; }; } // namespace jit } // namespace paddle