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