Skip to content

Commit 7933770

Browse files
larryliu0820facebook-github-bot
authored andcommitted
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
1 parent 8dc54d5 commit 7933770

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

extension/runner_util/managed_tensor.h

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,10 @@
1010
#include <executorch/runtime/core/exec_aten/util/dim_order_util.h>
1111
#include <executorch/runtime/core/exec_aten/util/tensor_util.h>
1212
#include <executorch/runtime/platform/assert.h>
13+
#include <memory>
14+
// NOTE: required by torchchat install_et.sh script.
15+
// @nolint PATTERNLINT Ok to use stdlib for this optional library
16+
#include <vector>
1317

1418
#ifdef USE_ATEN_LIB
1519
#include <torch/torch.h>

0 commit comments

Comments
 (0)