diff --git a/extension/runner_util/managed_tensor.h b/extension/runner_util/managed_tensor.h index aa4657bcb58..16b712d6595 100644 --- a/extension/runner_util/managed_tensor.h +++ b/extension/runner_util/managed_tensor.h @@ -10,6 +10,10 @@ #include #include #include +#include +// NOTE: required by torchchat install_et.sh script. +// @nolint PATTERNLINT Ok to use stdlib for this optional library +#include #ifdef USE_ATEN_LIB #include