From ec1efa3dd2a54be9623a77dfdf9015b78ce62992 Mon Sep 17 00:00:00 2001 From: Ryan Hill Date: Thu, 15 Apr 2021 14:41:58 -0700 Subject: [PATCH] ROCM --- onnxruntime/contrib_ops/cuda/math/isfinite.cc | 2 +- onnxruntime/core/providers/shared_library/provider_api.h | 4 +++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/onnxruntime/contrib_ops/cuda/math/isfinite.cc b/onnxruntime/contrib_ops/cuda/math/isfinite.cc index 121daa1880..0ebe21b23d 100644 --- a/onnxruntime/contrib_ops/cuda/math/isfinite.cc +++ b/onnxruntime/contrib_ops/cuda/math/isfinite.cc @@ -15,7 +15,7 @@ namespace cuda { 1, \ T, \ kCudaExecutionProvider, \ - KernelDefBuilder() \ + (*KernelDefBuilder::Create()) \ .TypeConstraint("V", DataTypeImpl::GetTensorType()) \ .TypeConstraint("T", DataTypeImpl::GetTensorType()), \ IsAllFiniteOp); diff --git a/onnxruntime/core/providers/shared_library/provider_api.h b/onnxruntime/core/providers/shared_library/provider_api.h index b24536c765..44a8b5986a 100644 --- a/onnxruntime/core/providers/shared_library/provider_api.h +++ b/onnxruntime/core/providers/shared_library/provider_api.h @@ -7,7 +7,9 @@ // switching providers to be runnable as shared libraries. The interfaces will become more tightly integrated into the core code. #pragma once -#ifndef USE_ROCM +#ifdef USE_ROCM +#include "core/providers/common.h" +#else #define SHARED_PROVIDER 1 #include