@@ -695,6 +695,33 @@ inline constexpr auto c8 = ::executorch::aten::ScalarType::ComplexDouble;
695695inline constexpr auto b1 = ::executorch::aten::ScalarType::Bool;
696696inline constexpr auto bf = ::executorch::aten::ScalarType::BFloat16;
697697
698+ using U1 =
699+ typename ScalarTypeToCppType<::executorch::aten::ScalarType::Byte>::type;
700+ using I1 =
701+ typename ScalarTypeToCppType<::executorch::aten::ScalarType::Char>::type;
702+ using I2 =
703+ typename ScalarTypeToCppType<::executorch::aten::ScalarType::Short>::type;
704+ using I4 =
705+ typename ScalarTypeToCppType<::executorch::aten::ScalarType::Int>::type;
706+ using I8 =
707+ typename ScalarTypeToCppType<::executorch::aten::ScalarType::Long>::type;
708+ using F2 =
709+ typename ScalarTypeToCppType<::executorch::aten::ScalarType::Half>::type;
710+ using F4 =
711+ typename ScalarTypeToCppType<::executorch::aten::ScalarType::Float>::type;
712+ using F8 =
713+ typename ScalarTypeToCppType<::executorch::aten::ScalarType::Double>::type;
714+ using C2 = typename ScalarTypeToCppType<
715+ ::executorch::aten::ScalarType::ComplexHalf>::type;
716+ using C4 = typename ScalarTypeToCppType<
717+ ::executorch::aten::ScalarType::ComplexFloat>::type;
718+ using C8 = typename ScalarTypeToCppType<
719+ ::executorch::aten::ScalarType::ComplexDouble>::type;
720+ using B1 =
721+ typename ScalarTypeToCppType<::executorch::aten::ScalarType::Bool>::type;
722+ using BF = typename ScalarTypeToCppType<
723+ ::executorch::aten::ScalarType::BFloat16>::type;
724+
698725inline constexpr std::
699726 array<int , int (::executorch::aten::ScalarType::NumOptions)>
700727 dtype2index = {{
@@ -1293,5 +1320,19 @@ using ::executorch::runtime::toString;
12931320using ::executorch::runtime::is_floating_point;
12941321using ::executorch::runtime::is_reduced_floating_point;
12951322#endif
1323+ namespace internal {
1324+ using ::executorch::runtime::internal::B1;
1325+ using ::executorch::runtime::internal::C2;
1326+ using ::executorch::runtime::internal::C4;
1327+ using ::executorch::runtime::internal::C8;
1328+ using ::executorch::runtime::internal::F2;
1329+ using ::executorch::runtime::internal::F4;
1330+ using ::executorch::runtime::internal::F8;
1331+ using ::executorch::runtime::internal::I1;
1332+ using ::executorch::runtime::internal::I2;
1333+ using ::executorch::runtime::internal::I4;
1334+ using ::executorch::runtime::internal::I8;
1335+ using ::executorch::runtime::internal::U1;
1336+ } // namespace internal
12961337} // namespace executor
12971338} // namespace torch
0 commit comments