From dcafa85af27c64dae57d030d71b7b34112bb37ff Mon Sep 17 00:00:00 2001 From: Ryan Hill Date: Fri, 16 Apr 2021 18:04:40 -0700 Subject: [PATCH] Fix linux build --- onnxruntime/core/providers/cpu/controlflow/utils.h | 4 +++- onnxruntime/core/providers/cuda/controlflow/if.cc | 1 - 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/onnxruntime/core/providers/cpu/controlflow/utils.h b/onnxruntime/core/providers/cpu/controlflow/utils.h index 0f4d00152a..480f640a0c 100644 --- a/onnxruntime/core/providers/cpu/controlflow/utils.h +++ b/onnxruntime/core/providers/cpu/controlflow/utils.h @@ -8,7 +8,9 @@ #include #include "core/framework/feeds_fetches_manager.h" -#ifndef SHARED_PROVIDER +#ifdef SHARED_PROVIDER +#include "core/framework/ml_value.h" +#else #include "core/framework/op_kernel.h" #endif diff --git a/onnxruntime/core/providers/cuda/controlflow/if.cc b/onnxruntime/core/providers/cuda/controlflow/if.cc index 41a5baf082..f34ace718c 100644 --- a/onnxruntime/core/providers/cuda/controlflow/if.cc +++ b/onnxruntime/core/providers/cuda/controlflow/if.cc @@ -2,7 +2,6 @@ // Licensed under the MIT License. #include "core/providers/shared_library/provider_api.h" -#include "core/framework/ml_value.h" #include "core/providers/cuda/controlflow/if.h" #include "core/providers/cuda/cuda_fwd.h"