From 7cb566e39f9b2c5573bc747d60db89e24b297544 Mon Sep 17 00:00:00 2001 From: Mengwei Liu Date: Mon, 22 Apr 2024 10:51:29 -0700 Subject: [PATCH] Add memory and vector include in managed_tensor.h (#3201) Summary: In order to get rid of this patch https://github.com/pytorch/torchchat/blob/main/scripts/install_et.sh#L35-L36 We upstream the changes into ExecuTorch. Reviewed By: lucylq Differential Revision: D56424633 --- extension/runner_util/managed_tensor.h | 4 ++++ 1 file changed, 4 insertions(+) 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