Add OptionalIntArrayRef used by torchgen. (#6145)

Add OptionalIntArrayRef used by torchgen. (#5735) Summary: Pull Request resolved: #5735 Adds OptionalIntArrayRef for generated cpp using torchgen. Generated code with torchgen.api.types.CppSignatureGroup uses additional aliases that are not currently defined in exec_aten.h. This diff adds support for OptionalIntArrayRef. Example: aten::mean.dim uses an optional int array that would end up using `OptionalIntArrayRef`. Reviewed By: zonglinpeng Differential Revision: D63568016 fbshipit-source-id: fc87338632c5d2e84f1dd9b088d69ad304a13d1c (cherry picked from commit 0e5b92d) Co-authored-by: Matthias Cremon <[email protected]>
pytorch · Oct 11, 2024 · e0fcdd4 · e0fcdd4
1 parent 4ad5350
commit e0fcdd4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/runtime/core/exec_aten/exec_aten.h b/runtime/core/exec_aten/exec_aten.h
@@ -87,6 +87,7 @@ using IntArrayRef = at::IntArrayRef;
 
 template <typename T>
 using OptionalArrayRef = c10::OptionalArrayRef<T>;
+using OptionalIntArrayRef = OptionalArrayRef<int64_t>;
 
 inline ssize_t compute_numel(const SizesType* sizes, ssize_t dim) {
   return static_cast<ssize_t>(
@@ -132,6 +133,7 @@ using IntArrayRef = torch::executor::IntArrayRef;
 template <typename T>
 using OptionalArrayRef =
     torch::executor::optional<torch::executor::ArrayRef<T>>;
+using OptionalIntArrayRef = OptionalArrayRef<int64_t>;
 
 using torch::executor::compute_numel;