Skip to content

Commit

Permalink
Add memory and vector include in managed_tensor.h (#3201)
Browse files Browse the repository at this point in the history
Summary:
Pull Request resolved: #3201

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

fbshipit-source-id: 72e6675b467416753b0fd995d8e514396eef8331
  • Loading branch information
larryliu0820 authored and facebook-github-bot committed Apr 22, 2024
1 parent d24af2b commit 90d0c1a
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions extension/runner_util/managed_tensor.h
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,10 @@
#include <executorch/runtime/core/exec_aten/util/dim_order_util.h>
#include <executorch/runtime/core/exec_aten/util/tensor_util.h>
#include <executorch/runtime/platform/assert.h>
#include <memory>
// NOTE: required by torchchat install_et.sh script.
// @nolint PATTERNLINT Ok to use stdlib for this optional library
#include <vector>

#ifdef USE_ATEN_LIB
#include <torch/torch.h>
Expand Down

0 comments on commit 90d0c1a

Please sign in to comment.