forked from huawei/mindspore2022
!20000 [MSLITE][Develop] move lite api: delegate, kernel to mindspore/include
Merge pull request !20000 from yangruoqi713/r1.3
This commit is contained in:
commit
cd2d2bef0a
|
|
@ -21,7 +21,7 @@
|
|||
#include <vector>
|
||||
#include <memory>
|
||||
#include "schema/model_generated.h"
|
||||
#include "include/kernel.h"
|
||||
#include "include/api/kernel.h"
|
||||
|
||||
namespace mindspore {
|
||||
typedef enum {
|
||||
|
|
@ -24,7 +24,7 @@
|
|||
#include "schema/model_generated.h"
|
||||
#include "include/api/context.h"
|
||||
#include "include/api/types.h"
|
||||
#include "include/kernel.h"
|
||||
#include "include/api/kernel.h"
|
||||
#include "ir/dtype/type_id.h"
|
||||
|
||||
namespace mindspore {
|
||||
|
|
|
|||
|
|
@ -14,7 +14,7 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include "include/delegate.h"
|
||||
#include "include/api/delegate.h"
|
||||
namespace mindspore {
|
||||
const schema::Primitive *DelegateModel::GetPrimitive(kernel::Kernel *kernel) const {
|
||||
if (primitives_.find(kernel) != primitives_.end()) {
|
||||
|
|
|
|||
|
|
@ -17,7 +17,7 @@
|
|||
#define MINDSPORE_LITE_SRC_DELEGATE_DELEGATE_UTILS
|
||||
#include <vector>
|
||||
#include "include/ms_tensor.h"
|
||||
#include "include/delegate.h"
|
||||
#include "include/api/delegate.h"
|
||||
#include "src/common/log_adapter.h"
|
||||
#include "src/delegate/tensorrt/op/tensorrt_op.h"
|
||||
|
||||
|
|
|
|||
|
|
@ -19,7 +19,7 @@
|
|||
|
||||
#include <vector>
|
||||
#include <map>
|
||||
#include "include/delegate.h"
|
||||
#include "include/api/delegate.h"
|
||||
#include "include/context.h"
|
||||
#include "src/delegate/npu/npu_manager.h"
|
||||
#include "src/delegate/npu/pass/npu_pass_manager.h"
|
||||
|
|
|
|||
|
|
@ -20,7 +20,7 @@
|
|||
#include <vector>
|
||||
#include <map>
|
||||
#include <utility>
|
||||
#include "include/kernel.h"
|
||||
#include "include/api/kernel.h"
|
||||
#include "src/delegate/npu/op/npu_op.h"
|
||||
#include "src/delegate/npu/npu_executor.h"
|
||||
|
||||
|
|
|
|||
|
|
@ -20,7 +20,7 @@
|
|||
#include <memory>
|
||||
#include <vector>
|
||||
#include <string>
|
||||
#include "include/kernel.h"
|
||||
#include "include/api/kernel.h"
|
||||
#include "src/delegate/npu/npu_executor.h"
|
||||
|
||||
namespace mindspore {
|
||||
|
|
|
|||
|
|
@ -19,7 +19,7 @@
|
|||
#include <vector>
|
||||
#include <string>
|
||||
#include "include/graph/op/all_ops.h"
|
||||
#include "include/kernel.h"
|
||||
#include "include/api/kernel.h"
|
||||
#include "include/errorcode.h"
|
||||
#include "src/common/log_adapter.h"
|
||||
|
||||
|
|
|
|||
|
|
@ -20,7 +20,7 @@
|
|||
#include <NvInfer.h>
|
||||
#include <string>
|
||||
#include <vector>
|
||||
#include "include/kernel.h"
|
||||
#include "include/api/kernel.h"
|
||||
#include "src/common/log_adapter.h"
|
||||
#include "include/errorcode.h"
|
||||
|
||||
|
|
|
|||
|
|
@ -18,9 +18,9 @@
|
|||
#include <string>
|
||||
#include <vector>
|
||||
#include <map>
|
||||
#include "include/delegate.h"
|
||||
#include "include/api/delegate.h"
|
||||
#include "src/delegate/tensorrt/tensorrt_subgraph.h"
|
||||
#include "include/kernel.h"
|
||||
#include "include/api/kernel.h"
|
||||
#include "include/errorcode.h"
|
||||
#include "src/common/log_adapter.h"
|
||||
|
||||
|
|
|
|||
|
|
@ -19,7 +19,7 @@
|
|||
#include <set>
|
||||
#include <string>
|
||||
#include <vector>
|
||||
#include "include/kernel.h"
|
||||
#include "include/api/kernel.h"
|
||||
#include "src/delegate/tensorrt/tensorrt_runtime.h"
|
||||
#include "src/delegate/tensorrt/tensorrt_utils.h"
|
||||
|
||||
|
|
|
|||
|
|
@ -30,7 +30,7 @@
|
|||
#include "schema/model_generated.h"
|
||||
#include "src/cxx_api/tensor/tensor_impl.h"
|
||||
#include "include/api/context.h"
|
||||
#include "include/kernel.h"
|
||||
#include "include/api/kernel.h"
|
||||
|
||||
namespace mindspore::kernel {
|
||||
class InnerKernel : public Kernel {
|
||||
|
|
|
|||
|
|
@ -32,10 +32,10 @@
|
|||
#include "include/errorcode.h"
|
||||
#include "schema/model_generated.h"
|
||||
#include "include/context.h"
|
||||
#include "include/kernel.h"
|
||||
#include "include/api/kernel.h"
|
||||
#include "src/cxx_api/tensor/tensor_impl.h"
|
||||
#include "src/inner_kernel.h"
|
||||
#include "include/delegate.h"
|
||||
#include "include/api/delegate.h"
|
||||
|
||||
namespace mindspore::kernel {
|
||||
enum KERNEL_ARCH { kCPU, kGPU, kAPU, kNPU, kCustom, kDelegate, kKernelArch_MIN = kCPU, kKernelArch_MAX = kAPU };
|
||||
|
|
|
|||
|
|
@ -31,7 +31,7 @@
|
|||
#include "src/executor.h"
|
||||
#include "src/tensor.h"
|
||||
#include "src/tensorlist.h"
|
||||
#include "include/delegate.h"
|
||||
#include "include/api/delegate.h"
|
||||
#if GPU_OPENCL
|
||||
#include "src/runtime/gpu/opencl/opencl_runtime.h"
|
||||
#elif GPU_VULKAN
|
||||
|
|
|
|||
|
|
@ -28,7 +28,7 @@
|
|||
#include "src/inner_context.h"
|
||||
#include "include/model.h"
|
||||
#include "src/scheduler_cb.h"
|
||||
#include "include/delegate.h"
|
||||
#include "include/api/delegate.h"
|
||||
|
||||
namespace mindspore::lite {
|
||||
class Scheduler {
|
||||
|
|
|
|||
Loading…
Reference in New Issue