From 9c6cc88e828b9913ac18b23ab2a889d4f61c2a9d Mon Sep 17 00:00:00 2001 From: Ryan Hill Date: Thu, 15 Apr 2021 15:18:39 -0700 Subject: [PATCH] ROCM --- onnxruntime/core/providers/shared_library/provider_api.h | 1 + 1 file changed, 1 insertion(+) diff --git a/onnxruntime/core/providers/shared_library/provider_api.h b/onnxruntime/core/providers/shared_library/provider_api.h index feea6e9cae..4b842e0f7d 100644 --- a/onnxruntime/core/providers/shared_library/provider_api.h +++ b/onnxruntime/core/providers/shared_library/provider_api.h @@ -10,6 +10,7 @@ #ifdef USE_ROCM #include "core/providers/common.h" #include "core/providers/cpu/tensor/onehot.h" +#include "core/providers/cpu/tensor/gather_elements.h" #else #define SHARED_PROVIDER 1