From fa577218089d32a3617abf5e594cf16e2730e330 Mon Sep 17 00:00:00 2001 From: Ryan Hill Date: Wed, 14 Apr 2021 19:37:53 -0700 Subject: [PATCH] Test fix for ROCM provider --- onnxruntime/core/providers/shared_library/provider_api.h | 3 +++ 1 file changed, 3 insertions(+) diff --git a/onnxruntime/core/providers/shared_library/provider_api.h b/onnxruntime/core/providers/shared_library/provider_api.h index 421f14cbcd..08fc1547c5 100644 --- a/onnxruntime/core/providers/shared_library/provider_api.h +++ b/onnxruntime/core/providers/shared_library/provider_api.h @@ -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 @@ -298,3 +299,5 @@ constexpr ONNXTensorElementDataType GetONNXTensorElementDataType() { r #define LOGS_DEFAULT(severity) \ LOGS_DEFAULT_CATEGORY(severity, ::onnxruntime::logging::Category::onnxruntime) + +#endif