saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-01 03:45:06 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Test fix for ROCM provider

Browse source
This commit is contained in:
Ryan Hill 2021-04-14 19:37:53 -07:00
parent 80cae23393
commit fa57721808
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -7,6 +7,7 @@
// switching providers to be runnable as shared libraries. The interfaces will become more tightly integrated into the core code.
#pragma once
#ifndef USE_ROCM
#define SHARED_PROVIDER 1
#include <vector>
@ -298,3 +299,5 @@ constexpr ONNXTensorElementDataType GetONNXTensorElementDataType<uint64_t>() { r
#define LOGS_DEFAULT(severity) \
LOGS_DEFAULT_CATEGORY(severity, ::onnxruntime::logging::Category::onnxruntime)
#endif