saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-26 19:52:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

ROCM Build Test

Browse source
This commit is contained in:
Ryan Hill 2021-04-15 14:13:58 -07:00
parent ae38f05886
commit 88f51f0ecf
2 changed files with 3 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
include/onnxruntime/core/framework/kernel_def_builder.h
View file

@ -180,6 +180,8 @@ class KernelDef {
class KernelDefBuilder {
public:
std::unique_ptr<KernelDefBuilder> Create() { return onnxruntime::make_unique<KernelDefBuilder>(); }
explicit KernelDefBuilder()
: kernel_def_(new KernelDef()) {}

17
onnxruntime/core/providers/shared_library/provider_api.h
View file

@ -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 <vector>
#include <string>
#include <map>
#include <gsl/gsl>
#include <unordered_map>
#include <unordered_set>
#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 <vector>