From 88f51f0ecf11a8577f393a1ab4aba9770ef83e34 Mon Sep 17 00:00:00 2001 From: Ryan Hill Date: Thu, 15 Apr 2021 14:13:58 -0700 Subject: [PATCH] ROCM Build Test --- .../core/framework/kernel_def_builder.h | 2 ++ .../providers/shared_library/provider_api.h | 17 +---------------- 2 files changed, 3 insertions(+), 16 deletions(-) diff --git a/include/onnxruntime/core/framework/kernel_def_builder.h b/include/onnxruntime/core/framework/kernel_def_builder.h index fa71276e8b..0b3bfb00a3 100644 --- a/include/onnxruntime/core/framework/kernel_def_builder.h +++ b/include/onnxruntime/core/framework/kernel_def_builder.h @@ -180,6 +180,8 @@ class KernelDef { class KernelDefBuilder { public: + std::unique_ptr Create() { return onnxruntime::make_unique(); } + explicit KernelDefBuilder() : kernel_def_(new KernelDef()) {} diff --git a/onnxruntime/core/providers/shared_library/provider_api.h b/onnxruntime/core/providers/shared_library/provider_api.h index e6949cb635..b24536c765 100644 --- a/onnxruntime/core/providers/shared_library/provider_api.h +++ b/onnxruntime/core/providers/shared_library/provider_api.h @@ -7,22 +7,7 @@ // switching providers to be runnable as shared libraries. The interfaces will become more tightly integrated into the core code. #pragma once -#ifdef USE_ROCM -#include -#include -#include -#include -#include -#include -#include "onnx/common/stl_backports.h" -#include "core/common/common.h" -#include "core/framework/data_transfer.h" -#include "core/framework/execution_provider.h" -#include "core/framework/op_kernel.h" -#include "core/framework/data_types_internal.h" -#include "core/framework/tensorprotoutils.h" -#include "core/providers/common.h" -#else +#ifndef USE_ROCM #define SHARED_PROVIDER 1 #include