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