// Copyright (c) 2019 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 #include "NeuronAdapter.h" #include "lite/core/kernel.h" #include "lite/core/subgraph_bridge_registry.h" #include "lite/core/subgraph_engine_base.h" namespace paddle { namespace lite { namespace kernels { namespace apu { class DeviceProgram { public: DeviceProgram() {} ~DeviceProgram() {} std::string GenerateModelName( const std::vector& input_names, const std::vector& output_names, const std::vector>& origin_idims); bool LoadFromCacheFile(const std::vector& input_names, const std::vector& output_names, const std::vector>& origin_idims, const std::string& model_cache_dir); bool BuildGraphAndCacheToFile( RuntimeProgram* origin_program, const std::vector& input_names, const std::vector& output_names, const std::vector>& origin_idims, const std::vector& origin_itensors, const std::vector& origin_otensors, const std::string& model_cache_dir); public: std::string model_name_{""}; std::vector> origin_odims_; std::vector origin_otypes_; NeuronModel* model_; NeuronCompilation* compilation_; }; class SubgraphEngine : public subgraph::SubgraphEngineBase { public: SubgraphEngine(KernelContext* ctx, int block_idx, const std::shared_ptr& program_desc, Scope* exec_scope, const std::vector& input_names, const std::vector& output_names) : subgraph::SubgraphEngineBase(ctx, block_idx, program_desc, exec_scope, input_names, output_names) {} ~SubgraphEngine(); protected: bool BuildDeviceProgram() override; bool LaunchDeviceProgram() override; std::map>, std::shared_ptr> device_programs_; }; class SubgraphCompute : public KernelLite { public: using param_t = operators::SubgraphParam; void PrepareForRun() override; void Run() override; virtual ~SubgraphCompute() = default; private: std::unique_ptr engine_; }; } // namespace apu } // namespace kernels } // namespace lite } // namespace paddle