react-native-executorch 0.4.8 → 0.5.1-rc.0

package/third-party/include/executorch/runtime/core/exec_aten/util/scalar_type_util.h

@@ -0,0 +1,1331 @@
+/*
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
+ * All rights reserved.
+ *
+ * This source code is licensed under the BSD-style license found in the
+ * LICENSE file in the root directory of this source tree.
+ */
+
+/**
+ * @file
+ *
+ * Forked from
+ * https://github.com/pytorch/pytorch/blob/master/c10/core/ScalarType.h
+ *
+ * See file comment in ../ScalarType.h.
+ *
+ * This file contains all of the non-critical parts of the original ScalarType.h
+ * that are not required for the core ExecuTorch runtime, but may be helpful for
+ * code that uses ScalarType.
+ */
+
+#pragma once
+
+#include <array>
+#include <cinttypes>
+#include <cstdint>
+#include <limits>
+#include <type_traits>
+
+#include <executorch/runtime/platform/assert.h>
+
+#ifdef USE_ATEN_LIB
+// Note that a lot of the macros/functions defined in this ScalarTypeUtil.h file
+// are also defined in c10/core/ScalarType.h, which is included via
+// kernel_types.h when building in ATen mode. They tend to use different names
+// and a different namespace, but if there are conflicts they should be resolved
+// here.
+#define ET_FORALL_SCALAR_TYPES AT_FORALL_SCALAR_TYPES_WITH_COMPLEX_AND_QINTS
+#include <c10/core/ScalarType.h>
+namespace executorch {
+namespace aten {
+using ScalarType = at::ScalarType;
+} // namespace aten
+} // namespace executorch
+#else // !USE_ATEN_LIB
+#include <executorch/runtime/core/portable_type/scalar_type.h>
+#include <executorch/runtime/core/portable_type/string_view.h>
+namespace executorch {
+namespace aten {
+using ScalarType = torch::executor::ScalarType;
+using string_view = torch::executor::string_view;
+} // namespace aten
+} // namespace executorch
+#endif // USE_ATEN_LIB
+// DEPRECATED: The executorch::aten:: namespace is deprecated. Use
+// executorch::aten:: instead.
+namespace exec_aten = ::executorch::aten;
+
+namespace executorch {
+namespace runtime {
+
+#if !defined(USE_ATEN_LIB)
+// Util to figure out if the scalar type if one of the
+// supported floating point types.
+// In aten mode, aten lib already has these utils as part of
+// its vec_base.h
+template <typename T>
+struct is_floating_point
+    : std::integral_constant<bool,
+                             std::is_floating_point<T>::value ||
+                                 std::is_same_v<T, torch::executor::Half> ||
+                                 std::is_same_v<T, torch::executor::BFloat16>> {
+};
+
+// Util to figure out if the scalar type is one of the
+// reduced precision floating point types.
+template <typename T>
+struct is_reduced_floating_point
+    : std::integral_constant<bool,
+                             std::is_same_v<T, torch::executor::Half> ||
+                                 std::is_same_v<T, torch::executor::BFloat16>> {
+};
+
+template <typename T>
+constexpr bool is_reduced_floating_point_v =
+    is_reduced_floating_point<T>::value;
+#endif
+
+/// Maps ScalarTypes to C++ types.
+template <::executorch::aten::ScalarType N> struct ScalarTypeToCppType;
+
+#define SPECIALIZE_ScalarTypeToCppType(cpp_type, scalar_type)                  \
+  template <>                                                                  \
+  struct ScalarTypeToCppType<::executorch::aten::ScalarType::scalar_type> {    \
+    using type = cpp_type;                                                     \
+  };
+
+ET_FORALL_SCALAR_TYPES(SPECIALIZE_ScalarTypeToCppType)
+
+#undef SPECIALIZE_ScalarTypeToCppType
+
+/// Maps C++ types to ScalarTypes.
+template <typename T> struct CppTypeToScalarType;
+
+#define SPECIALIZE_CppTypeToScalarType(cpp_type, scalar_type)                  \
+  template <>                                                                  \
+  struct CppTypeToScalarType<cpp_type>                                         \
+      : std::integral_constant<::executorch::aten::ScalarType,                 \
+                               ::executorch::aten::ScalarType::scalar_type> {  \
+  };
+
+ET_FORALL_SCALAR_TYPES(SPECIALIZE_CppTypeToScalarType)
+
+#undef SPECIALIZE_CppTypeToScalarType
+
+//
+// Macros that iterate across different subsets of ScalarTypes.
+//
+// See ET_FORALL_SCALAR_TYPES in ScalarType.h to iterate across all ScalarType
+// names and types.
+//
+// For all of these macros, the final `_` parameter is the name of another macro
+// that takes two parameters: the name of a C type, and the name of the
+// corresponding ScalarType enumerator.
+//
+// Note that these macros should use fully-qualified namespaces (starting with
+// `::`) to ensure that they can be called safely in any arbitrary namespace.
+//
+
+// In this context, "INT" means integer C types, which is why the quantized
+// integer types are not included.
+#define ET_FORALL_INT_TYPES(_)                                                 \
+  _(uint8_t, Byte)                                                             \
+  _(int8_t, Char)                                                              \
+  _(int16_t, Short)                                                            \
+  _(int32_t, Int)                                                              \
+  _(int64_t, Long)
+
+// Here `ANOTHER_INPUT` should be another variable to be forwarded to a given
+// function.
+#define ET_FORALL_INT_TYPES_WITH(ANOTHER_INPUT, _)                             \
+  _(ANOTHER_INPUT, uint8_t, Byte)                                              \
+  _(ANOTHER_INPUT, int8_t, Char)                                               \
+  _(ANOTHER_INPUT, int16_t, Short)                                             \
+  _(ANOTHER_INPUT, int32_t, Int)                                               \
+  _(ANOTHER_INPUT, int64_t, Long)
+
+#define ET_FORALL_INT_TYPES_WITH2(ANOTHER_INPUT1, ANOTHER_INPUT2, _)           \
+  _(ANOTHER_INPUT1, ANOTHER_INPUT2, uint8_t, Byte)                             \
+  _(ANOTHER_INPUT1, ANOTHER_INPUT2, int8_t, Char)                              \
+  _(ANOTHER_INPUT1, ANOTHER_INPUT2, int16_t, Short)                            \
+  _(ANOTHER_INPUT1, ANOTHER_INPUT2, int32_t, Int)                              \
+  _(ANOTHER_INPUT1, ANOTHER_INPUT2, int64_t, Long)
+
+#define ET_FORALL_INT_TYPES_AND(SCALARTYPE, _)                                 \
+  _(uint8_t, Byte)                                                             \
+  _(int8_t, Char)                                                              \
+  _(int16_t, Short)                                                            \
+  _(int32_t, Int)                                                              \
+  _(int64_t, Long)                                                             \
+  _(::executorch::runtime::ScalarTypeToCppType<                                \
+        ::executorch::aten::ScalarType::SCALARTYPE>::type,                     \
+    SCALARTYPE)
+
+// In this context, "FLOAT" means float C types, which is why BFloat16 is not
+// included.
+#define ET_FORALL_FLOAT_TYPES(_)                                               \
+  _(float, Float)                                                              \
+  _(double, Double)
+
+#define ET_FORALL_FLOAT_TYPES_AND(SCALARTYPE, _)                               \
+  _(float, Float)                                                              \
+  _(double, Double)                                                            \
+  _(::executorch::runtime::ScalarTypeToCppType<                                \
+        ::executorch::aten::ScalarType::SCALARTYPE>::type,                     \
+    SCALARTYPE)
+
+#define ET_FORALL_FLOAT_TYPES_AND2(SCALARTYPE1, SCALARTYPE2, _)                \
+  _(float, Float)                                                              \
+  _(double, Double)                                                            \
+  _(::executorch::runtime::ScalarTypeToCppType<                                \
+        ::executorch::aten::ScalarType::SCALARTYPE1>::type,                    \
+    SCALARTYPE1)                                                               \
+  _(::executorch::runtime::ScalarTypeToCppType<                                \
+        ::executorch::aten::ScalarType::SCALARTYPE2>::type,                    \
+    SCALARTYPE2)
+
+#define ET_FORALL_FLOATH_TYPES(_) ET_FORALL_FLOAT_TYPES_AND(Half, _)
+
+#define ET_FORALL_FLOATHBF16_TYPES(_)                                          \
+  ET_FORALL_FLOAT_TYPES_AND2(Half, BFloat16, _)
+
+// Here `ANOTHER_INPUT` should be another variable to be forwarded to a given
+// function. Not to be confused with another scalar type as in
+// `ET_FORALL_FLOAT_TYPES_AND`.
+#define ET_FORALL_FLOAT_TYPES_WITH(ANOTHER_INPUT, _)                           \
+  _(ANOTHER_INPUT, float, Float)                                               \
+  _(ANOTHER_INPUT, double, Double)
+
+#define ET_FORALL_FLOAT_TYPES_WITH2(ANOTHER_INPUT1, ANOTHER_INPUT2, _)         \
+  _(ANOTHER_INPUT1, ANOTHER_INPUT2, float, Float)                              \
+  _(ANOTHER_INPUT1, ANOTHER_INPUT2, double, Double)
+
+#define ET_FORALL_FLOATHBF16_TYPES_WITH2(ANOTHER_INPUT1, ANOTHER_INPUT2, _)    \
+  _(ANOTHER_INPUT1, ANOTHER_INPUT2, float, Float)                              \
+  _(ANOTHER_INPUT1, ANOTHER_INPUT2, double, Double)                            \
+  _(ANOTHER_INPUT1, ANOTHER_INPUT2, ::executorch::aten::Half, Half)            \
+  _(ANOTHER_INPUT1, ANOTHER_INPUT2, ::executorch::aten::BFloat16, BFloat16)
+
+// In this context, "REAL" means integer/float C types, which is why BFloat16
+// and Half are not included.
+#define ET_FORALL_REAL_TYPES(_)                                                \
+  _(uint8_t, Byte)                                                             \
+  _(int8_t, Char)                                                              \
+  _(int16_t, Short)                                                            \
+  _(int32_t, Int)                                                              \
+  _(int64_t, Long)                                                             \
+  _(float, Float)                                                              \
+  _(double, Double)
+
+// Here `ANOTHER_INPUT` should be another variable to be forwarded to a given
+// function. Not to be confused with another scalar type as in
+// `ET_FORALL_REAL_TYPES_AND`.
+#define ET_FORALL_REAL_TYPES_WITH(ANOTHER_INPUT, _)                            \
+  _(ANOTHER_INPUT, uint8_t, Byte)                                              \
+  _(ANOTHER_INPUT, int8_t, Char)                                               \
+  _(ANOTHER_INPUT, int16_t, Short)                                             \
+  _(ANOTHER_INPUT, int32_t, Int)                                               \
+  _(ANOTHER_INPUT, int64_t, Long)                                              \
+  _(ANOTHER_INPUT, float, Float)                                               \
+  _(ANOTHER_INPUT, double, Double)
+
+#define ET_FORALL_REAL_TYPES_WITH2(ANOTHER_INPUT1, ANOTHER_INPUT2, _)          \
+  _(ANOTHER_INPUT1, ANOTHER_INPUT2, uint8_t, Byte)                             \
+  _(ANOTHER_INPUT1, ANOTHER_INPUT2, int8_t, Char)                              \
+  _(ANOTHER_INPUT1, ANOTHER_INPUT2, int16_t, Short)                            \
+  _(ANOTHER_INPUT1, ANOTHER_INPUT2, int32_t, Int)                              \
+  _(ANOTHER_INPUT1, ANOTHER_INPUT2, int64_t, Long)                             \
+  _(ANOTHER_INPUT1, ANOTHER_INPUT2, float, Float)                              \
+  _(ANOTHER_INPUT1, ANOTHER_INPUT2, double, Double)
+
+#define ET_FORALL_REALHBF16_TYPES_WITH2(ANOTHER_INPUT1, ANOTHER_INPUT2, _)     \
+  _(ANOTHER_INPUT1, ANOTHER_INPUT2, uint8_t, Byte)                             \
+  _(ANOTHER_INPUT1, ANOTHER_INPUT2, int8_t, Char)                              \
+  _(ANOTHER_INPUT1, ANOTHER_INPUT2, int16_t, Short)                            \
+  _(ANOTHER_INPUT1, ANOTHER_INPUT2, int32_t, Int)                              \
+  _(ANOTHER_INPUT1, ANOTHER_INPUT2, int64_t, Long)                             \
+  _(ANOTHER_INPUT1, ANOTHER_INPUT2, float, Float)                              \
+  _(ANOTHER_INPUT1, ANOTHER_INPUT2, double, Double)                            \
+  _(ANOTHER_INPUT1, ANOTHER_INPUT2, ::executorch::aten::Half, Half)            \
+  _(ANOTHER_INPUT1, ANOTHER_INPUT2, ::executorch::aten::BFloat16, BFloat16)
+
+#define ET_FORALL_REALHBBF16_TYPES_WITH2(ANOTHER_INPUT1, ANOTHER_INPUT2, _)    \
+  ET_FORALL_REALHBF16_TYPES_WITH2(ANOTHER_INPUT2, ANOTHER_INPUT2, _)           \
+  _(ANOTHER_INPUT1, ANOTHER_INPUT2, bool, Bool)
+
+// For macros that take `SCALARTYPEn` parameters, those parameters should be
+// an unquoted/unqualified enumerator name like `Int` or `Float`.
+#define ET_FORALL_REAL_TYPES_AND(SCALARTYPE, _)                                \
+  _(uint8_t, Byte)                                                             \
+  _(int8_t, Char)                                                              \
+  _(int16_t, Short)                                                            \
+  _(int32_t, Int)                                                              \
+  _(int64_t, Long)                                                             \
+  _(float, Float)                                                              \
+  _(double, Double)                                                            \
+  _(::executorch::runtime::ScalarTypeToCppType<                                \
+        ::executorch::aten::ScalarType::SCALARTYPE>::type,                     \
+    SCALARTYPE)
+
+#define ET_FORALL_REAL_TYPES_AND2(SCALARTYPE1, SCALARTYPE2, _)                 \
+  _(uint8_t, Byte)                                                             \
+  _(int8_t, Char)                                                              \
+  _(int16_t, Short)                                                            \
+  _(int32_t, Int)                                                              \
+  _(int64_t, Long)                                                             \
+  _(float, Float)                                                              \
+  _(double, Double)                                                            \
+  _(::executorch::runtime::ScalarTypeToCppType<                                \
+        ::executorch::aten::ScalarType::SCALARTYPE1>::type,                    \
+    SCALARTYPE1)                                                               \
+  _(::executorch::runtime::ScalarTypeToCppType<                                \
+        ::executorch::aten::ScalarType::SCALARTYPE2>::type,                    \
+    SCALARTYPE2)
+
+#define ET_FORALL_REALH_TYPES(_) ET_FORALL_REAL_TYPES_AND(Half, _)
+
+#define ET_FORALL_REALHBF16_TYPES(_)                                           \
+  ET_FORALL_REAL_TYPES_AND2(Half, BFloat16, _)
+
+#define ET_FORALL_REALHBBF16_TYPES(_)                                          \
+  ET_FORALL_REAL_TYPES_AND3(Bool, Half, BFloat16, _)
+
+#define ET_FORALL_REAL_TYPES_AND_WITH(SCALARTYPE, ANOTHER_INPUT, _)            \
+  _(ANOTHER_INPUT, uint8_t, Byte)                                              \
+  _(ANOTHER_INPUT, int8_t, Char)                                               \
+  _(ANOTHER_INPUT, int16_t, Short)                                             \
+  _(ANOTHER_INPUT, int32_t, Int)                                               \
+  _(ANOTHER_INPUT, int64_t, Long)                                              \
+  _(ANOTHER_INPUT, float, Float)                                               \
+  _(ANOTHER_INPUT, double, Double)                                             \
+  _(ANOTHER_INPUT,                                                             \
+    ::executorch::runtime::ScalarTypeToCppType<                                \
+        ::executorch::aten::ScalarType::SCALARTYPE>::type,                     \
+    SCALARTYPE)
+
+#define ET_FORALL_REAL_TYPES_AND2(SCALARTYPE1, SCALARTYPE2, _)                 \
+  _(uint8_t, Byte)                                                             \
+  _(int8_t, Char)                                                              \
+  _(int16_t, Short)                                                            \
+  _(int32_t, Int)                                                              \
+  _(int64_t, Long)                                                             \
+  _(float, Float)                                                              \
+  _(double, Double)                                                            \
+  _(::executorch::runtime::ScalarTypeToCppType<                                \
+        ::executorch::aten::ScalarType::SCALARTYPE1>::type,                    \
+    SCALARTYPE1)                                                               \
+  _(::executorch::runtime::ScalarTypeToCppType<                                \
+        ::executorch::aten::ScalarType::SCALARTYPE2>::type,                    \
+    SCALARTYPE2)
+
+#define ET_FORALL_REAL_TYPES_AND3(SCALARTYPE1, SCALARTYPE2, SCALARTYPE3, _)    \
+  _(uint8_t, Byte)                                                             \
+  _(int8_t, Char)                                                              \
+  _(int16_t, Short)                                                            \
+  _(int32_t, Int)                                                              \
+  _(int64_t, Long)                                                             \
+  _(float, Float)                                                              \
+  _(double, Double)                                                            \
+  _(::executorch::runtime::ScalarTypeToCppType<                                \
+        ::executorch::aten::ScalarType::SCALARTYPE1>::type,                    \
+    SCALARTYPE1)                                                               \
+  _(::executorch::runtime::ScalarTypeToCppType<                                \
+        ::executorch::aten::ScalarType::SCALARTYPE2>::type,                    \
+    SCALARTYPE2)                                                               \
+  _(::executorch::runtime::ScalarTypeToCppType<                                \
+        ::executorch::aten::ScalarType::SCALARTYPE3>::type,                    \
+    SCALARTYPE3)
+
+#define ET_FORALL_QINT_TYPES(_)                                                \
+  _(::torch::executor::qint8, QInt8)                                           \
+  _(::torch::executor::quint8, QUInt8)                                         \
+  _(::torch::executor::qint32, QInt32)                                         \
+  _(::torch::executor::quint4x2, QUInt4x2)                                     \
+  _(::torch::executor::quint2x4, QUInt2x4)
+
+// In this context, "COMPLEX" means complex types based on primitive C types,
+// which is why ComplexHalf is not included.
+#define ET_FORALL_COMPLEX_TYPES(_)                                             \
+  _(::executorch::aten::complex<float>, ComplexFloat)                          \
+  _(::executorch::aten::complex<double>, ComplexDouble)
+
+#define ET_FORALL_COMPLEXH_TYPES(_)                                            \
+  _(::executorch::aten::complex<::executorch::aten::Half>, ComplexHalf)        \
+  _(::executorch::aten::complex<float>, ComplexFloat)                          \
+  _(::executorch::aten::complex<double>, ComplexDouble)
+
+//
+// Utility functions to retrieve metadata for a given ScalarType
+//
+
+/**
+ * Returns true if the parameter is one of the values covered by
+ * ET_FORALL_SCALAR_TYPES.
+ */
+inline bool isValid(::executorch::aten::ScalarType type) {
+  return static_cast<int8_t>(type) >= 0 &&
+         type < ::executorch::aten::ScalarType::NumOptions &&
+         type != ::executorch::aten::ScalarType::Undefined;
+}
+
+/**
+ * Returns the name of a ScalarType as a C string.
+ *
+ * @param[in] t The type to get the name of.
+ * @return The name of the type, or "UNKNOWN_SCALAR" if the type is not known.
+ */
+inline const char *toString(::executorch::aten::ScalarType t) {
+#define DEFINE_CASE(_, name)                                                   \
+  case ::executorch::aten::ScalarType::name:                                   \
+    return #name;
+
+  switch (t) {
+    ET_FORALL_SCALAR_TYPES(DEFINE_CASE)
+  case ::executorch::aten::ScalarType::Undefined:
+    return "Undefined";
+  default:
+    return "UNKNOWN_SCALAR";
+  }
+#undef DEFINE_CASE
+}
+
+/**
+ * Returns the size in bytes of the C type associated with the ScalarType.
+ *
+ * Calls ET_CHECK_MSG() if the type is unknown or is ScalarType::Undefined.
+ *
+ * @param[in] t The type to get the underlying C type size of.
+ * @return The size of the associated C type in bytes.
+ */
+inline size_t elementSize(::executorch::aten::ScalarType t) {
+#define CASE_ELEMENTSIZE_CASE(ctype, name)                                     \
+  case ::executorch::aten::ScalarType::name:                                   \
+    return sizeof(ctype);
+
+  switch (t) {
+    ET_FORALL_SCALAR_TYPES(CASE_ELEMENTSIZE_CASE)
+  default:
+    ET_CHECK_MSG(false, "Unknown ScalarType %" PRId8, static_cast<int8_t>(t));
+  }
+#undef CASE_ELEMENTSIZE_CASE
+}
+
+inline constexpr bool isIntegralType(::executorch::aten::ScalarType t,
+                                     bool includeBool) {
+  return (includeBool && t == ::executorch::aten::ScalarType::Bool) ||
+         (t == ::executorch::aten::ScalarType::Byte ||
+          t == ::executorch::aten::ScalarType::Char ||
+          t == ::executorch::aten::ScalarType::Int ||
+          t == ::executorch::aten::ScalarType::Long ||
+          t == ::executorch::aten::ScalarType::Short);
+}
+
+template <typename T, bool includeBool>
+struct is_integral_type
+    : public std::integral_constant<
+          bool, isIntegralType(CppTypeToScalarType<T>::value, includeBool)> {};
+
+inline constexpr bool isFloatingType(::executorch::aten::ScalarType t) {
+  return (t == ::executorch::aten::ScalarType::Double ||
+          t == ::executorch::aten::ScalarType::Float ||
+          t == ::executorch::aten::ScalarType::Half ||
+          t == ::executorch::aten::ScalarType::BFloat16);
+}
+
+inline bool isRealType(::executorch::aten::ScalarType t) {
+  return (t == ::executorch::aten::ScalarType::Byte ||
+          t == ::executorch::aten::ScalarType::Char ||
+          t == ::executorch::aten::ScalarType::Short ||
+          t == ::executorch::aten::ScalarType::Int ||
+          t == ::executorch::aten::ScalarType::Long ||
+          t == ::executorch::aten::ScalarType::Float ||
+          t == ::executorch::aten::ScalarType::Double);
+}
+
+inline bool isRealHType(::executorch::aten::ScalarType t) {
+  return (t == ::executorch::aten::ScalarType::Byte ||
+          t == ::executorch::aten::ScalarType::Char ||
+          t == ::executorch::aten::ScalarType::Short ||
+          t == ::executorch::aten::ScalarType::Int ||
+          t == ::executorch::aten::ScalarType::Long ||
+          t == ::executorch::aten::ScalarType::Float ||
+          t == ::executorch::aten::ScalarType::Double ||
+          t == ::executorch::aten::ScalarType::Half);
+}
+
+inline bool isRealHBType(::executorch::aten::ScalarType t) {
+  return (isRealHType(t) || t == ::executorch::aten::ScalarType::Bool);
+}
+
+inline bool isRealHBF16Type(::executorch::aten::ScalarType t) {
+  return (isRealHType(t) || t == ::executorch::aten::ScalarType::BFloat16);
+}
+
+inline bool isRealHBBF16Type(::executorch::aten::ScalarType t) {
+  return (isRealHBType(t) || t == ::executorch::aten::ScalarType::BFloat16);
+}
+
+inline constexpr bool isComplexType(::executorch::aten::ScalarType t) {
+  return (t == ::executorch::aten::ScalarType::ComplexHalf ||
+          t == ::executorch::aten::ScalarType::ComplexFloat ||
+          t == ::executorch::aten::ScalarType::ComplexDouble);
+}
+
+template <typename T>
+struct is_complex_type
+    : std::integral_constant<bool,
+                             isComplexType(CppTypeToScalarType<T>::value)> {};
+
+constexpr bool isBitsType(::executorch::aten::ScalarType t) {
+  return t == ::executorch::aten::ScalarType::Bits1x8 ||
+         t == ::executorch::aten::ScalarType::Bits2x4 ||
+         t == ::executorch::aten::ScalarType::Bits4x2 ||
+         t == ::executorch::aten::ScalarType::Bits8 ||
+         t == ::executorch::aten::ScalarType::Bits16;
+}
+
+template <typename T>
+struct is_bits_type
+    : std::integral_constant<bool, isBitsType(CppTypeToScalarType<T>::value)> {
+};
+
+constexpr bool isQIntType(::executorch::aten::ScalarType t) {
+  // Don't forget to extend this when adding new QInt types
+  return t == ::executorch::aten::ScalarType::QInt8 ||
+         t == ::executorch::aten::ScalarType::QUInt8 ||
+         t == ::executorch::aten::ScalarType::QInt32 ||
+         t == ::executorch::aten::ScalarType::QUInt4x2 ||
+         t == ::executorch::aten::ScalarType::QUInt2x4;
+}
+
+template <typename T>
+struct is_qint_type
+    : std::integral_constant<bool, isQIntType(CppTypeToScalarType<T>::value)> {
+};
+
+constexpr bool isFloat8Type(::executorch::aten::ScalarType t) {
+  // Don't forget to extend this when adding new QInt types
+  return t == ::executorch::aten::ScalarType::Float8_e5m2 ||
+         t == ::executorch::aten::ScalarType::Float8_e4m3fn ||
+         t == ::executorch::aten::ScalarType::Float8_e5m2fnuz ||
+         t == ::executorch::aten::ScalarType::Float8_e4m3fnuz;
+}
+
+template <typename T>
+struct is_float8_type
+    : std::integral_constant<bool,
+                             isFloat8Type(CppTypeToScalarType<T>::value)> {};
+
+constexpr bool isBarebonesUnsignedType(::executorch::aten::ScalarType t) {
+  // Don't forget to extend this when adding new QInt types
+  return t == ::executorch::aten::ScalarType::UInt16 ||
+         t == ::executorch::aten::ScalarType::UInt32 ||
+         t == ::executorch::aten::ScalarType::UInt64;
+}
+
+template <typename T>
+struct is_barebones_unsigned_type
+    : std::integral_constant<bool, isBarebonesUnsignedType(
+                                       CppTypeToScalarType<T>::value)> {};
+
+inline ::executorch::aten::ScalarType
+toQIntType(::executorch::aten::ScalarType t) {
+  switch (t) {
+  case ::executorch::aten::ScalarType::Byte:
+    return ::executorch::aten::ScalarType::QUInt8;
+  case ::executorch::aten::ScalarType::Char:
+    return ::executorch::aten::ScalarType::QInt8;
+  case ::executorch::aten::ScalarType::Int:
+    return ::executorch::aten::ScalarType::QInt32;
+  default:
+    return t;
+  }
+}
+
+inline ::executorch::aten::ScalarType
+toUnderlying(::executorch::aten::ScalarType t) {
+  switch (t) {
+  case ::executorch::aten::ScalarType::QUInt8:
+    return ::executorch::aten::ScalarType::Byte;
+  case ::executorch::aten::ScalarType::QInt8:
+    return ::executorch::aten::ScalarType::Char;
+  case ::executorch::aten::ScalarType::QInt32:
+    return ::executorch::aten::ScalarType::Int;
+  case ::executorch::aten::ScalarType::QUInt4x2:
+    return ::executorch::aten::ScalarType::Byte;
+  case ::executorch::aten::ScalarType::QUInt2x4:
+    return ::executorch::aten::ScalarType::Byte;
+  default:
+    return t;
+  }
+}
+
+inline bool isSignedType(::executorch::aten::ScalarType t) {
+  ET_CHECK_MSG(!::executorch::runtime::isQIntType(t),
+               "isSignedType not supported for quantized types like %" PRId8,
+               static_cast<int8_t>(t));
+#define CASE_SIGNED(ctype, name)                                               \
+  case ::executorch::aten::ScalarType::name:                                   \
+    return std::numeric_limits<ctype>::is_signed;
+
+  switch (t) {
+  case ::executorch::aten::ScalarType::ComplexHalf:
+  case ::executorch::aten::ScalarType::ComplexFloat:
+  case ::executorch::aten::ScalarType::ComplexDouble:
+    return true;
+    ET_FORALL_REAL_TYPES_AND3(Half, Bool, BFloat16, CASE_SIGNED)
+  default:
+    ET_CHECK_MSG(false, "Unknown ScalarType %" PRId8, static_cast<int8_t>(t));
+  }
+#undef CASE_SIGNED
+}
+
+inline bool isUnderlying(::executorch::aten::ScalarType type,
+                         ::executorch::aten::ScalarType qtype) {
+  return type == ::executorch::runtime::toUnderlying(qtype);
+}
+
+inline constexpr ::executorch::aten::ScalarType
+toRealValueType(::executorch::aten::ScalarType t) {
+  switch (t) {
+  case ::executorch::aten::ScalarType::ComplexHalf:
+    return ::executorch::aten::ScalarType::Half;
+  case ::executorch::aten::ScalarType::ComplexFloat:
+    return ::executorch::aten::ScalarType::Float;
+  case ::executorch::aten::ScalarType::ComplexDouble:
+    return ::executorch::aten::ScalarType::Double;
+  default:
+    return t;
+  }
+}
+
+inline constexpr ::executorch::aten::ScalarType
+toComplexType(::executorch::aten::ScalarType t) {
+  switch (t) {
+  case ::executorch::aten::ScalarType::BFloat16:
+    // BFloat16 has range equivalent to Float,
+    // so we map it to ComplexFloat.
+    return ::executorch::aten::ScalarType::ComplexFloat;
+  case ::executorch::aten::ScalarType::Half:
+    return ::executorch::aten::ScalarType::ComplexHalf;
+  case ::executorch::aten::ScalarType::Float:
+    return ::executorch::aten::ScalarType::ComplexFloat;
+  case ::executorch::aten::ScalarType::Double:
+    return ::executorch::aten::ScalarType::ComplexDouble;
+  case ::executorch::aten::ScalarType::ComplexHalf:
+    return ::executorch::aten::ScalarType::ComplexHalf;
+  case ::executorch::aten::ScalarType::ComplexFloat:
+    return ::executorch::aten::ScalarType::ComplexFloat;
+  case ::executorch::aten::ScalarType::ComplexDouble:
+    return ::executorch::aten::ScalarType::ComplexDouble;
+  default:
+    ET_CHECK_MSG(false, "Unknown Complex ScalarType for %" PRId8,
+                 static_cast<int8_t>(t));
+  }
+}
+
+/**
+ * Encodes type casting rules that are consistent with ATen behaviour.
+ */
+inline constexpr bool canCast(const ::executorch::aten::ScalarType from,
+                              const ::executorch::aten::ScalarType to) {
+  // Disallow complex -> non-complex
+  return !(::executorch::runtime::isComplexType(from) &&
+           !::executorch::runtime::isComplexType(to)) &&
+         // Disallow float -> integral
+         !(::executorch::runtime::isFloatingType(from) &&
+           ::executorch::runtime::isIntegralType(to, /*includeBool=*/false)) &&
+         // Treat bool as a special category. Disallow non-bool -> bool
+         !(from != ::executorch::aten::ScalarType::Bool &&
+           to == ::executorch::aten::ScalarType::Bool);
+}
+
+template <typename T1, typename T2>
+struct can_cast
+    : std::integral_constant<bool, canCast(CppTypeToScalarType<T1>::value,
+                                           CppTypeToScalarType<T2>::value)> {};
+
+/**
+ * When casting from floating point to integral type, if the floating value is
+ * outside the integral type range, then an error is thrown if sanitization is
+ * enabled. To circumvent this, we cast the floating point to int64_t first.
+ */
+template <typename To, typename From,
+          std::enable_if_t<(std::is_floating_point<From>::value &&
+                            std::is_integral<To>::value),
+                           int> = 0>
+To convert(From val) {
+  return static_cast<To>(static_cast<int64_t>(val));
+}
+
+template <typename To, typename From,
+          std::enable_if_t<!(std::is_floating_point<From>::value &&
+                             std::is_integral<To>::value),
+                           int> = 0>
+To convert(From val) {
+  return static_cast<To>(val);
+}
+
+namespace internal {
+// This is generated according to NumPy's promote_types
+inline constexpr auto u1 = ::executorch::aten::ScalarType::Byte;
+inline constexpr auto i1 = ::executorch::aten::ScalarType::Char;
+inline constexpr auto i2 = ::executorch::aten::ScalarType::Short;
+inline constexpr auto i4 = ::executorch::aten::ScalarType::Int;
+inline constexpr auto i8 = ::executorch::aten::ScalarType::Long;
+inline constexpr auto f2 = ::executorch::aten::ScalarType::Half;
+inline constexpr auto f4 = ::executorch::aten::ScalarType::Float;
+inline constexpr auto f8 = ::executorch::aten::ScalarType::Double;
+inline constexpr auto c2 = ::executorch::aten::ScalarType::ComplexHalf;
+inline constexpr auto c4 = ::executorch::aten::ScalarType::ComplexFloat;
+inline constexpr auto c8 = ::executorch::aten::ScalarType::ComplexDouble;
+inline constexpr auto b1 = ::executorch::aten::ScalarType::Bool;
+inline constexpr auto bf = ::executorch::aten::ScalarType::BFloat16;
+
+using U1 =
+    typename ScalarTypeToCppType<::executorch::aten::ScalarType::Byte>::type;
+using I1 =
+    typename ScalarTypeToCppType<::executorch::aten::ScalarType::Char>::type;
+using I2 =
+    typename ScalarTypeToCppType<::executorch::aten::ScalarType::Short>::type;
+using I4 =
+    typename ScalarTypeToCppType<::executorch::aten::ScalarType::Int>::type;
+using I8 =
+    typename ScalarTypeToCppType<::executorch::aten::ScalarType::Long>::type;
+using F2 =
+    typename ScalarTypeToCppType<::executorch::aten::ScalarType::Half>::type;
+using F4 =
+    typename ScalarTypeToCppType<::executorch::aten::ScalarType::Float>::type;
+using F8 =
+    typename ScalarTypeToCppType<::executorch::aten::ScalarType::Double>::type;
+using C2 = typename ScalarTypeToCppType<
+    ::executorch::aten::ScalarType::ComplexHalf>::type;
+using C4 = typename ScalarTypeToCppType<
+    ::executorch::aten::ScalarType::ComplexFloat>::type;
+using C8 = typename ScalarTypeToCppType<
+    ::executorch::aten::ScalarType::ComplexDouble>::type;
+using B1 =
+    typename ScalarTypeToCppType<::executorch::aten::ScalarType::Bool>::type;
+using BF = typename ScalarTypeToCppType<
+    ::executorch::aten::ScalarType::BFloat16>::type;
+
+inline constexpr std::array<::executorch::aten::ScalarType, 13> index2dtype = {
+    {u1, i1, i2, i4, i8, f2, f4, f8, c2, c4, c8, b1, bf}};
+
+constexpr std::array<
+    int64_t, static_cast<size_t>(::executorch::aten::ScalarType::NumOptions)>
+calculate_dtype2index() {
+  std::array<int64_t,
+             static_cast<size_t>(::executorch::aten::ScalarType::NumOptions)>
+      inverse = {};
+  for (int64_t i = 0;
+       i < static_cast<int64_t>(::executorch::aten::ScalarType::NumOptions);
+       i++) {
+    inverse[i] = -1;
+  }
+  for (int64_t i = 0; i < static_cast<int64_t>(index2dtype.size()); i++) {
+    inverse[static_cast<int64_t>(index2dtype[i])] = i;
+  }
+  return inverse;
+}
+
+inline constexpr auto dtype2index = calculate_dtype2index();
+inline constexpr int NUM_PROMOTE_TYPES = 13;
+// Should match _promoteTypesLookup in c10/core/ScalarType.cpp so that
+// we match PyTorch core type promotion semantics.
+inline constexpr ::executorch::aten::ScalarType
+    promoteTypesLookup[NUM_PROMOTE_TYPES][NUM_PROMOTE_TYPES] = {
+        /*        u1  i1  i2  i4  i8  f2  f4  f8  c2  c4  c8  b1  bf*/
+        /* u1 */ {u1, i2, i2, i4, i8, f2, f4, f8, c2, c4, c8, u1, bf},
+        /* i1 */ {i2, i1, i2, i4, i8, f2, f4, f8, c2, c4, c8, i1, bf},
+        /* i2 */ {i2, i2, i2, i4, i8, f2, f4, f8, c2, c4, c8, i2, bf},
+        /* i4 */ {i4, i4, i4, i4, i8, f2, f4, f8, c2, c4, c8, i4, bf},
+        /* i8 */ {i8, i8, i8, i8, i8, f2, f4, f8, c2, c4, c8, i8, bf},
+        /* f2 */ {f2, f2, f2, f2, f2, f2, f4, f8, c2, c4, c8, f2, f4},
+        /* f4 */ {f4, f4, f4, f4, f4, f4, f4, f8, c4, c4, c8, f4, f4},
+        /* f8 */ {f8, f8, f8, f8, f8, f8, f8, f8, c8, c8, c8, f8, f8},
+        /* c2 */ {c2, c2, c2, c2, c2, c2, c4, c8, c2, c4, c8, c2, c4},
+        /* c4 */ {c4, c4, c4, c4, c4, c4, c4, c8, c4, c4, c8, c4, c4},
+        /* c8 */ {c8, c8, c8, c8, c8, c8, c8, c8, c8, c8, c8, c8, c8},
+        /* b1 */ {u1, i1, i2, i4, i8, f2, f4, f8, c2, c4, c8, b1, bf},
+        /* bf */ {bf, bf, bf, bf, bf, f4, f4, f8, c4, c4, c8, bf, bf},
+};
+
+} // namespace internal
+
+/**
+ * Implements type promotion rules that are consistent with ATen behaviour,
+ * which in turn is consistent with NumPy's promote_types.
+ * If half_to_float is set to true, then half and bfloat16 will be promoted to
+ * float instead
+ */
+inline constexpr ::executorch::aten::ScalarType
+promoteTypes(::executorch::aten::ScalarType a, ::executorch::aten::ScalarType b,
+             bool half_to_float = false) {
+  // For QInt types, only allow exact match
+  if (::executorch::runtime::isQIntType(a) && a == b) {
+    return a;
+  }
+  if (::executorch::runtime::isQIntType(a) ||
+      ::executorch::runtime::isQIntType(b)) {
+    ET_CHECK_MSG(false, "promoteTypes not valid for quantized dtypes");
+  }
+
+  // For Bits types, only allow exact match
+  if (::executorch::runtime::isBitsType(a) && a == b) {
+    return a;
+  }
+  if (::executorch::runtime::isBitsType(a) ||
+      ::executorch::runtime::isBitsType(b)) {
+    ET_CHECK_MSG(false, "promoteTypes not valid for bits dtypes");
+  }
+
+  // For Float8 types, only allow exact match
+  if (::executorch::runtime::isFloat8Type(a) && a == b) {
+    return a;
+  }
+  if (::executorch::runtime::isFloat8Type(a) ||
+      ::executorch::runtime::isFloat8Type(b)) {
+    ET_CHECK_MSG(false, "promoteTypes not valid for float8 dtypes");
+  }
+
+  // For barebones uint types, only allow exact match
+  if (::executorch::runtime::isBarebonesUnsignedType(a) && a == b) {
+    return a;
+  }
+  if (::executorch::runtime::isBarebonesUnsignedType(a) ||
+      ::executorch::runtime::isBarebonesUnsignedType(b)) {
+    ET_CHECK_MSG(false, "promoteTypes not valid for barebone unsigned dtypes");
+  }
+
+  auto ix_a = ::executorch::runtime::internal::dtype2index[(int)a];
+  ET_CHECK(ix_a != -1);
+  auto ix_b = ::executorch::runtime::internal::dtype2index[(int)b];
+  ET_CHECK(ix_b != -1);
+  ::executorch::aten::ScalarType promoted_type =
+      ::executorch::runtime::internal::promoteTypesLookup[ix_a][ix_b];
+
+  if (half_to_float &&
+      (promoted_type == ::executorch::aten::ScalarType::Half ||
+       promoted_type == ::executorch::aten::ScalarType::BFloat16)) {
+    promoted_type = ::executorch::aten::ScalarType::Float;
+  }
+
+  return promoted_type;
+}
+
+template <typename T1, typename T2, bool half_to_float = false>
+struct promote_types {
+private:
+  static_assert(std::is_same_v<T1, T2> ||
+                    (!is_qint_type<T1>::value && !is_qint_type<T2>::value),
+                "promote_types not valid for quantized dtypes");
+  static_assert(std::is_same_v<T1, T2> ||
+                    (!is_bits_type<T1>::value && !is_bits_type<T2>::value),
+                "promote_types not valid for bits dtypes");
+  static_assert(std::is_same_v<T1, T2> ||
+                    (!is_float8_type<T1>::value && !is_float8_type<T2>::value),
+                "promote_types not valid for float8 dtypes");
+  static_assert(std::is_same_v<T1, T2> ||
+                    (!is_barebones_unsigned_type<T1>::value &&
+                     !is_barebones_unsigned_type<T2>::value),
+                "promote_types not valid for barebones unsigned dtypes");
+
+  using promoted_type_not_respecting_half_to_float =
+      typename ScalarTypeToCppType<promoteTypes(
+          CppTypeToScalarType<T1>::value,
+          CppTypeToScalarType<T2>::value)>::type;
+
+public:
+  using type = std::conditional_t<
+      half_to_float &&
+          (std::is_same_v<promoted_type_not_respecting_half_to_float,
+                          typename ScalarTypeToCppType<
+                              ::executorch::aten::ScalarType::Half>::type> ||
+           std::is_same_v<promoted_type_not_respecting_half_to_float,
+                          typename ScalarTypeToCppType<
+                              ::executorch::aten::ScalarType::BFloat16>::type>),
+      typename ScalarTypeToCppType<::executorch::aten::ScalarType::Float>::type,
+      promoted_type_not_respecting_half_to_float>;
+};
+
+//
+// Helper macros for switch case macros (see below)
+//
+// These macros are not meant to be used directly. They provide an easy way to
+// generate a switch statement that can handle subsets of ScalarTypes supported
+// by ExecuTorch.
+//
+
+#ifdef ET_INTERNAL_CHECK_SELECTIVE_BUILD
+#define ET_INTERNAL_SWITCH_CASE(enum_type, CTYPE_ALIAS, ...)                   \
+  case enum_type: {                                                            \
+    ET_INTERNAL_CHECK_SELECTIVE_BUILD(enum_type);                              \
+    using CTYPE_ALIAS =                                                        \
+        ::executorch::runtime::ScalarTypeToCppType<enum_type>::type;           \
+    return __VA_ARGS__();                                                      \
+  }
+#else
+#define ET_INTERNAL_SWITCH_CASE(enum_type, CTYPE_ALIAS, ...)                   \
+  case enum_type: {                                                            \
+    using CTYPE_ALIAS =                                                        \
+        ::executorch::runtime::ScalarTypeToCppType<enum_type>::type;           \
+    return __VA_ARGS__();                                                      \
+  }
+#endif
+
+#define ET_INTERNAL_SWITCH(TYPE, CONTEXT, NAME, ...)                           \
+  [&] {                                                                        \
+    const auto &_st = TYPE;                                                    \
+    constexpr const char *et_switch_name = NAME;                               \
+    (void)et_switch_name; /* Suppress unused var */                            \
+    switch (_st) {                                                             \
+      __VA_ARGS__                                                              \
+    default:                                                                   \
+      ET_CHECK_MSG(false, "Unhandled dtype %s for %s",                         \
+                   ::executorch::runtime::toString(_st), et_switch_name);      \
+    }                                                                          \
+  }()
+
+#define ET_INTERNAL_SWITCH_CASE_ALL_TYPES(CTYPE_ALIAS, ...)                    \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Byte, CTYPE_ALIAS,   \
+                          __VA_ARGS__)                                         \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Char, CTYPE_ALIAS,   \
+                          __VA_ARGS__)                                         \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Short, CTYPE_ALIAS,  \
+                          __VA_ARGS__)                                         \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Int, CTYPE_ALIAS,    \
+                          __VA_ARGS__)                                         \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Long, CTYPE_ALIAS,   \
+                          __VA_ARGS__)                                         \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Half, CTYPE_ALIAS,   \
+                          __VA_ARGS__)                                         \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Float, CTYPE_ALIAS,  \
+                          __VA_ARGS__)                                         \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Double, CTYPE_ALIAS, \
+                          __VA_ARGS__)                                         \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::ComplexHalf,         \
+                          CTYPE_ALIAS, __VA_ARGS__)                            \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::ComplexFloat,        \
+                          CTYPE_ALIAS, __VA_ARGS__)                            \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::ComplexDouble,       \
+                          CTYPE_ALIAS, __VA_ARGS__)                            \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Bool, CTYPE_ALIAS,   \
+                          __VA_ARGS__)                                         \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::QInt8, CTYPE_ALIAS,  \
+                          __VA_ARGS__)                                         \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::QUInt8, CTYPE_ALIAS, \
+                          __VA_ARGS__)                                         \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::QInt32, CTYPE_ALIAS, \
+                          __VA_ARGS__)                                         \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::BFloat16,            \
+                          CTYPE_ALIAS, __VA_ARGS__)                            \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::QUInt4x2,            \
+                          CTYPE_ALIAS, __VA_ARGS__)                            \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::QUInt2x4,            \
+                          CTYPE_ALIAS, __VA_ARGS__)                            \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Bits1x8,             \
+                          CTYPE_ALIAS, __VA_ARGS__)                            \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Bits2x4,             \
+                          CTYPE_ALIAS, __VA_ARGS__)                            \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Bits4x2,             \
+                          CTYPE_ALIAS, __VA_ARGS__)                            \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Bits8, CTYPE_ALIAS,  \
+                          __VA_ARGS__)                                         \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Bits16, CTYPE_ALIAS, \
+                          __VA_ARGS__)
+
+#define ET_INTERNAL_SWITCH_CASE_REAL_TYPES(CTYPE_ALIAS, ...)                   \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Byte, CTYPE_ALIAS,   \
+                          __VA_ARGS__)                                         \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Char, CTYPE_ALIAS,   \
+                          __VA_ARGS__)                                         \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Short, CTYPE_ALIAS,  \
+                          __VA_ARGS__)                                         \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Int, CTYPE_ALIAS,    \
+                          __VA_ARGS__)                                         \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Long, CTYPE_ALIAS,   \
+                          __VA_ARGS__)                                         \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Float, CTYPE_ALIAS,  \
+                          __VA_ARGS__)                                         \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Double, CTYPE_ALIAS, \
+                          __VA_ARGS__)
+
+#define ET_INTERNAL_SWITCH_CASE_REAL_TYPES_AND(ADDITIONAL, CTYPE_ALIAS, ...)   \
+  ET_INTERNAL_SWITCH_CASE_REAL_TYPES(CTYPE_ALIAS, __VA_ARGS__)                 \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::ADDITIONAL,          \
+                          CTYPE_ALIAS, __VA_ARGS__)
+
+#define ET_INTERNAL_SWITCH_CASE_REAL_TYPES_AND2(ADDITIONAL1, ADDITIONAL2,      \
+                                                CTYPE_ALIAS, ...)              \
+  ET_INTERNAL_SWITCH_CASE_REAL_TYPES(CTYPE_ALIAS, __VA_ARGS__)                 \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::ADDITIONAL1,         \
+                          CTYPE_ALIAS, __VA_ARGS__)                            \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::ADDITIONAL2,         \
+                          CTYPE_ALIAS, __VA_ARGS__)
+
+#define ET_INTERNAL_SWITCH_CASE_REAL_TYPES_AND3(ADDITIONAL1, ADDITIONAL2,      \
+                                                ADDITIONAL3, CTYPE_ALIAS, ...) \
+  ET_INTERNAL_SWITCH_CASE_REAL_TYPES(CTYPE_ALIAS, __VA_ARGS__)                 \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::ADDITIONAL1,         \
+                          CTYPE_ALIAS, __VA_ARGS__)                            \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::ADDITIONAL2,         \
+                          CTYPE_ALIAS, __VA_ARGS__)                            \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::ADDITIONAL3,         \
+                          CTYPE_ALIAS, __VA_ARGS__)
+
+#define ET_INTERNAL_SWITCH_CASE_INT_TYPES(CTYPE_ALIAS, ...)                    \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Byte, CTYPE_ALIAS,   \
+                          __VA_ARGS__)                                         \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Char, CTYPE_ALIAS,   \
+                          __VA_ARGS__)                                         \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Short, CTYPE_ALIAS,  \
+                          __VA_ARGS__)                                         \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Int, CTYPE_ALIAS,    \
+                          __VA_ARGS__)                                         \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Long, CTYPE_ALIAS,   \
+                          __VA_ARGS__)
+
+#define ET_INTERNAL_SWITCH_CASE_INT_TYPES_AND(ADDITIONAL, CTYPE_ALIAS, ...)    \
+  ET_INTERNAL_SWITCH_CASE_INT_TYPES(CTYPE_ALIAS, __VA_ARGS__)                  \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::ADDITIONAL,          \
+                          CTYPE_ALIAS, __VA_ARGS__)
+
+#define ET_INTERNAL_SWITCH_CASE_FLOAT_TYPES(CTYPE_ALIAS, ...)                  \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Double, CTYPE_ALIAS, \
+                          __VA_ARGS__)                                         \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Float, CTYPE_ALIAS,  \
+                          __VA_ARGS__)
+
+#define ET_INTERNAL_SWITCH_CASE_FLOAT_TYPES_AND(ADDITIONAL, CTYPE_ALIAS, ...)  \
+  ET_INTERNAL_SWITCH_CASE_FLOAT_TYPES(CTYPE_ALIAS, __VA_ARGS__)                \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::ADDITIONAL,          \
+                          CTYPE_ALIAS, __VA_ARGS__)
+
+#define ET_INTERNAL_SWITCH_CASE_FLOAT_TYPES_AND2(ADDITIONAL1, ADDITIONAL2,     \
+                                                 CTYPE_ALIAS, ...)             \
+  ET_INTERNAL_SWITCH_CASE_FLOAT_TYPES_AND(ADDITIONAL1, CTYPE_ALIAS,            \
+                                          __VA_ARGS__)                         \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::ADDITIONAL2,         \
+                          CTYPE_ALIAS, __VA_ARGS__)
+
+#define ET_INTERNAL_SWITCH_CASE_FLOAT_TYPES_AND3(                              \
+    ADDITIONAL1, ADDITIONAL2, ADDITIONAL3, CTYPE_ALIAS, ...)                   \
+  ET_INTERNAL_SWITCH_CASE_FLOAT_TYPES_AND2(ADDITIONAL1, ADDITIONAL2,           \
+                                           CTYPE_ALIAS, __VA_ARGS__)           \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::ADDITIONAL3,         \
+                          CTYPE_ALIAS, __VA_ARGS__)
+
+#define ET_INTERNAL_SWITCH_CASE_QINT_TYPES(CTYPE_ALIAS, ...)                   \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::QInt8, CTYPE_ALIAS,  \
+                          __VA_ARGS__)                                         \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::QUInt8, CTYPE_ALIAS, \
+                          __VA_ARGS__)                                         \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::QInt32, CTYPE_ALIAS, \
+                          __VA_ARGS__)                                         \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::QUInt4x2,            \
+                          CTYPE_ALIAS, __VA_ARGS__)                            \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::QUInt2x4,            \
+                          CTYPE_ALIAS, __VA_ARGS__)
+
+#define ET_INTERNAL_SWITCH_CASE_COMPLEX_TYPES(CTYPE_ALIAS, ...)                \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::ComplexFloat,        \
+                          CTYPE_ALIAS, __VA_ARGS__)                            \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::ComplexDouble,       \
+                          CTYPE_ALIAS, __VA_ARGS__)
+
+#define ET_INTERNAL_SWITCH_CASE_COMPLEXH_TYPES(CTYPE_ALIAS, ...)               \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::ComplexHalf,         \
+                          CTYPE_ALIAS, __VA_ARGS__)                            \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::ComplexFloat,        \
+                          CTYPE_ALIAS, __VA_ARGS__)                            \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::ComplexDouble,       \
+                          CTYPE_ALIAS, __VA_ARGS__)
+
+#define ET_INTERNAL_SWITCH_CASE_SCALAR_OBJ_TYPES(CTYPE_ALIAS, ...)             \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Bool, CTYPE_ALIAS,   \
+                          __VA_ARGS__)                                         \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Long, CTYPE_ALIAS,   \
+                          __VA_ARGS__)                                         \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Double, CTYPE_ALIAS, \
+                          __VA_ARGS__)
+
+#define ET_INTERNAL_SWITCH_CASE_SCALAR_OBJ_REAL_TYPES(CTYPE_ALIAS, ...)        \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Long, CTYPE_ALIAS,   \
+                          __VA_ARGS__)                                         \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Double, CTYPE_ALIAS, \
+                          __VA_ARGS__)
+
+#define ET_INTERNAL_SWITCH_CASE_SCALAR_OBJ_INTB_TYPES(CTYPE_ALIAS, ...)        \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Bool, CTYPE_ALIAS,   \
+                          __VA_ARGS__)                                         \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Long, CTYPE_ALIAS,   \
+                          __VA_ARGS__)
+
+#define ET_INTERNAL_SWITCH_CASE_SCALAR_OBJ_FLOATB_TYPES(CTYPE_ALIAS, ...)      \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Bool, CTYPE_ALIAS,   \
+                          __VA_ARGS__)                                         \
+  ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::Double, CTYPE_ALIAS, \
+                          __VA_ARGS__)
+
+//
+// Switch case macros
+//
+// These macros provide an easy way to generate switch statements that apply a
+// common lambda function to subsets of ScalarTypes supported by ExecuTorch.
+// The lambda function can type specialize to the ctype associated with the
+// ScalarType being handled through an alias passed as the CTYPE_ALIAS argument.
+//
+// Arguments:
+//   - ADDITIONAL: Additional ScalarType case to add
+//   - TYPE: The ScalarType to handle through the switch statement
+//   - CONTEXT: The KernelRuntimeContext instance used for error handling, etc.
+//   - NAME: A name for this operation which will be used in error messages
+//   - CTYPE_ALIAS: A typedef for the ctype associated with the ScalarType.
+//   - [&](){...}: A lambda function to be applied to each ScalarType case
+//
+// An example usage is:
+//
+// ET_SWITCH_REAL_TYPES(input.scalar_type(), "example", CTYPE, [&]() {
+//   output.mutable_data_ptr<CTYPE>[0] = input.const_data_ptr<CTYPE>[0];
+// });
+//
+// Note that these can be nested as well:
+//
+// ET_SWITCH_REAL_TYPES(input.scalar_type(), "example", CTYPE_IN, [&]() {
+//   ET_SWITCH_REAL_TYPES(output.scalar_type(), "example", CTYPE_OUT, [&]() {
+//     output.mutable_data_ptr<CTYPE_OUT>[0] =
+//         input.const_data_ptr<CTYPE_IN>[0];
+//   });
+// });
+//
+// These macros are adapted from Dispatch.h in the ATen library. The primary
+// difference is that the CTYPE_ALIAS argument is exposed to users, which is
+// used to alias the ctype associated with the ScalarType that is being handled.
+//
+
+#define ET_SWITCH_ALL_TYPES(TYPE, CONTEXT, NAME, CTYPE_ALIAS, ...)             \
+  ET_INTERNAL_SWITCH(                                                          \
+      TYPE, CONTEXT, NAME,                                                     \
+      ET_INTERNAL_SWITCH_CASE_ALL_TYPES(CTYPE_ALIAS, __VA_ARGS__))
+
+#define ET_SWITCH_REAL_TYPES(TYPE, CONTEXT, NAME, CTYPE_ALIAS, ...)            \
+  ET_INTERNAL_SWITCH(                                                          \
+      TYPE, CONTEXT, NAME,                                                     \
+      ET_INTERNAL_SWITCH_CASE_REAL_TYPES(CTYPE_ALIAS, __VA_ARGS__))
+
+#define ET_SWITCH_REAL_TYPES_AND(ADDITIONAL, TYPE, CONTEXT, NAME, CTYPE_ALIAS, \
+                                 ...)                                          \
+  ET_INTERNAL_SWITCH(TYPE, CONTEXT, NAME,                                      \
+                     ET_INTERNAL_SWITCH_CASE_REAL_TYPES_AND(                   \
+                         ADDITIONAL, CTYPE_ALIAS, __VA_ARGS__))
+
+#define ET_SWITCH_REAL_TYPES_AND2(ADDITIONAL1, ADDITIONAL2, TYPE, CONTEXT,     \
+                                  NAME, CTYPE_ALIAS, ...)                      \
+  ET_INTERNAL_SWITCH(TYPE, CONTEXT, NAME,                                      \
+                     ET_INTERNAL_SWITCH_CASE_REAL_TYPES_AND2(                  \
+                         ADDITIONAL1, ADDITIONAL2, CTYPE_ALIAS, __VA_ARGS__))
+
+#define ET_SWITCH_REAL_TYPES_AND3(ADDITIONAL1, ADDITIONAL2, ADDITIONAL3, TYPE, \
+                                  CONTEXT, NAME, CTYPE_ALIAS, ...)             \
+  ET_INTERNAL_SWITCH(                                                          \
+      TYPE, CONTEXT, NAME,                                                     \
+      ET_INTERNAL_SWITCH_CASE_REAL_TYPES_AND3(                                 \
+          ADDITIONAL1, ADDITIONAL2, ADDITIONAL3, CTYPE_ALIAS, __VA_ARGS__))
+
+#define ET_SWITCH_REALH_TYPES(TYPE, CONTEXT, NAME, CTYPE_ALIAS, ...)           \
+  ET_SWITCH_REAL_TYPES_AND(Half, TYPE, CONTEXT, NAME, CTYPE_ALIAS, __VA_ARGS__)
+
+#define ET_SWITCH_REALHBF16_TYPES(TYPE, CONTEXT, NAME, CTYPE_ALIAS, ...)       \
+  ET_SWITCH_REAL_TYPES_AND2(Half, BFloat16, TYPE, CONTEXT, NAME, CTYPE_ALIAS,  \
+                            __VA_ARGS__)
+
+#define ET_SWITCH_REALB_TYPES(TYPE, CONTEXT, NAME, CTYPE_ALIAS, ...)           \
+  ET_SWITCH_REAL_TYPES_AND(Bool, TYPE, CONTEXT, NAME, CTYPE_ALIAS, __VA_ARGS__)
+
+#define ET_SWITCH_REALHB_TYPES(TYPE, CONTEXT, NAME, CTYPE_ALIAS, ...)          \
+  ET_SWITCH_REAL_TYPES_AND2(Half, Bool, TYPE, CONTEXT, NAME, CTYPE_ALIAS,      \
+                            __VA_ARGS__)
+
+#define ET_SWITCH_REALHBBF16_TYPES(TYPE, CONTEXT, NAME, CTYPE_ALIAS, ...)      \
+  ET_SWITCH_REAL_TYPES_AND3(Half, Bool, BFloat16, TYPE, CONTEXT, NAME,         \
+                            CTYPE_ALIAS, __VA_ARGS__)
+
+#define ET_SWITCH_REALHBBF16_AND_UINT_TYPES(TYPE, CONTEXT, NAME, CTYPE_ALIAS,  \
+                                            ...)                               \
+  ET_INTERNAL_SWITCH(                                                          \
+      TYPE, CONTEXT, NAME,                                                     \
+      ET_INTERNAL_SWITCH_CASE_REAL_TYPES_AND3(Half, Bool, BFloat16,            \
+                                              CTYPE_ALIAS, __VA_ARGS__)        \
+          ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::UInt16,      \
+                                  CTYPE_ALIAS, __VA_ARGS__)                    \
+              ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::UInt32,  \
+                                      CTYPE_ALIAS, __VA_ARGS__)                \
+                  ET_INTERNAL_SWITCH_CASE(                                     \
+                      ::executorch::aten::ScalarType::UInt64, CTYPE_ALIAS,     \
+                      __VA_ARGS__))
+
+#define ET_SWITCH_INT_TYPES(TYPE, CONTEXT, NAME, CTYPE_ALIAS, ...)             \
+  ET_INTERNAL_SWITCH(                                                          \
+      TYPE, CONTEXT, NAME,                                                     \
+      ET_INTERNAL_SWITCH_CASE_INT_TYPES(CTYPE_ALIAS, __VA_ARGS__))
+
+#define ET_SWITCH_INT_TYPES_AND(ADDITIONAL, TYPE, CONTEXT, NAME, CTYPE_ALIAS,  \
+                                ...)                                           \
+  ET_INTERNAL_SWITCH(TYPE, CONTEXT, NAME,                                      \
+                     ET_INTERNAL_SWITCH_CASE_INT_TYPES_AND(                    \
+                         ADDITIONAL, CTYPE_ALIAS, __VA_ARGS__))
+
+#define ET_SWITCH_FLOAT_TYPES(TYPE, CONTEXT, NAME, CTYPE_ALIAS, ...)           \
+  ET_INTERNAL_SWITCH(                                                          \
+      TYPE, CONTEXT, NAME,                                                     \
+      ET_INTERNAL_SWITCH_CASE_FLOAT_TYPES(CTYPE_ALIAS, __VA_ARGS__))
+
+#define ET_SWITCH_FLOAT_TYPES_AND(ADDITIONAL, TYPE, CONTEXT, NAME,             \
+                                  CTYPE_ALIAS, ...)                            \
+  ET_INTERNAL_SWITCH(TYPE, CONTEXT, NAME,                                      \
+                     ET_INTERNAL_SWITCH_CASE_FLOAT_TYPES_AND(                  \
+                         ADDITIONAL, CTYPE_ALIAS, __VA_ARGS__))
+
+#define ET_SWITCH_FLOAT_TYPES_AND2(ADDITIONAL1, ADDITIONAL2, TYPE, CONTEXT,    \
+                                   NAME, CTYPE_ALIAS, ...)                     \
+  ET_INTERNAL_SWITCH(TYPE, CONTEXT, NAME,                                      \
+                     ET_INTERNAL_SWITCH_CASE_FLOAT_TYPES_AND2(                 \
+                         ADDITIONAL1, ADDITIONAL2, CTYPE_ALIAS, __VA_ARGS__))
+
+#define ET_SWITCH_FLOAT_TYPES_AND3(ADDITIONAL1, ADDITIONAL2, ADDITIONAL3,      \
+                                   TYPE, CONTEXT, NAME, CTYPE_ALIAS, ...)      \
+  ET_INTERNAL_SWITCH(                                                          \
+      TYPE, CONTEXT, NAME,                                                     \
+      ET_INTERNAL_SWITCH_CASE_FLOAT_TYPES_AND3(                                \
+          ADDITIONAL1, ADDITIONAL2, ADDITIONAL3, CTYPE_ALIAS, __VA_ARGS__))
+
+#define ET_SWITCH_FLOATH_TYPES(TYPE, CONTEXT, NAME, CTYPE_ALIAS, ...)          \
+  ET_SWITCH_FLOAT_TYPES_AND(Half, TYPE, CONTEXT, NAME, CTYPE_ALIAS, __VA_ARGS__)
+
+#define ET_SWITCH_FLOATHBF16_TYPES(TYPE, CONTEXT, NAME, CTYPE_ALIAS, ...)      \
+  ET_SWITCH_FLOAT_TYPES_AND2(Half, BFloat16, TYPE, CONTEXT, NAME, CTYPE_ALIAS, \
+                             __VA_ARGS__)
+
+#define ET_SWITCH_FLOATHBF16_TYPES_AND(ADDITIONAL, TYPE, CONTEXT, NAME,        \
+                                       CTYPE_ALIAS, ...)                       \
+  ET_SWITCH_FLOAT_TYPES_AND3(Half, BFloat16, ADDITIONAL, TYPE, CONTEXT, NAME,  \
+                             CTYPE_ALIAS, __VA_ARGS__)
+
+#define ET_SWITCH_QINT_TYPES(TYPE, CONTEXT, NAME, CTYPE_ALIAS, ...)            \
+  ET_INTERNAL_SWITCH(                                                          \
+      TYPE, CONTEXT, NAME,                                                     \
+      ET_INTERNAL_SWITCH_CASE_QINT_TYPES(CTYPE_ALIAS, __VA_ARGS__))
+
+#define ET_SWITCH_COMPLEX_TYPES(TYPE, CONTEXT, NAME, CTYPE_ALIAS, ...)         \
+  ET_INTERNAL_SWITCH(                                                          \
+      TYPE, CONTEXT, NAME,                                                     \
+      ET_INTERNAL_SWITCH_CASE_COMPLEX_TYPES(CTYPE_ALIAS, __VA_ARGS__))
+
+#define ET_SWITCH_COMPLEXH_TYPES(TYPE, CONTEXT, NAME, CTYPE_ALIAS, ...)        \
+  ET_INTERNAL_SWITCH(                                                          \
+      TYPE, CONTEXT, NAME,                                                     \
+      ET_INTERNAL_SWITCH_CASE_COMPLEXH_TYPES(CTYPE_ALIAS, __VA_ARGS__))
+
+#define ET_SWITCH_SCALAR_OBJ_TYPES(TYPE, CONTEXT, NAME, CTYPE_ALIAS, ...)      \
+  ET_INTERNAL_SWITCH(                                                          \
+      TYPE, CONTEXT, NAME,                                                     \
+      ET_INTERNAL_SWITCH_CASE_SCALAR_OBJ_TYPES(CTYPE_ALIAS, __VA_ARGS__))
+
+#define ET_SWITCH_SCALAR_OBJ_REAL_TYPES(TYPE, CONTEXT, NAME, CTYPE_ALIAS, ...) \
+  ET_INTERNAL_SWITCH(                                                          \
+      TYPE, CONTEXT, NAME,                                                     \
+      ET_INTERNAL_SWITCH_CASE_SCALAR_OBJ_REAL_TYPES(CTYPE_ALIAS, __VA_ARGS__))
+
+#define ET_SWITCH_SCALAR_OBJ_INTB_TYPES(TYPE, CONTEXT, NAME, CTYPE_ALIAS, ...) \
+  ET_INTERNAL_SWITCH(                                                          \
+      TYPE, CONTEXT, NAME,                                                     \
+      ET_INTERNAL_SWITCH_CASE_SCALAR_OBJ_INTB_TYPES(CTYPE_ALIAS, __VA_ARGS__))
+
+#define ET_SWITCH_SCALAR_OBJ_FLOATB_TYPES(TYPE, CONTEXT, NAME, CTYPE_ALIAS,    \
+                                          ...)                                 \
+  ET_INTERNAL_SWITCH(TYPE, CONTEXT, NAME,                                      \
+                     ET_INTERNAL_SWITCH_CASE_SCALAR_OBJ_FLOATB_TYPES(          \
+                         CTYPE_ALIAS, __VA_ARGS__))
+
+#define ET_SWITCH_TWO_TYPES(T1, T2, TYPE, CONTEXT, NAME, CTYPE_ALIAS, ...)     \
+  ET_INTERNAL_SWITCH(                                                          \
+      TYPE, CONTEXT, NAME,                                                     \
+      ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::T1, CTYPE_ALIAS, \
+                              __VA_ARGS__)                                     \
+          ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::T2,          \
+                                  CTYPE_ALIAS, __VA_ARGS__))
+
+#define ET_SWITCH_THREE_TYPES(T1, T2, T3, TYPE, CONTEXT, NAME, CTYPE_ALIAS,    \
+                              ...)                                             \
+  ET_INTERNAL_SWITCH(                                                          \
+      TYPE, CONTEXT, NAME,                                                     \
+      ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::T1, CTYPE_ALIAS, \
+                              __VA_ARGS__)                                     \
+          ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::T2,          \
+                                  CTYPE_ALIAS, __VA_ARGS__)                    \
+              ET_INTERNAL_SWITCH_CASE(::executorch::aten::ScalarType::T3,      \
+                                      CTYPE_ALIAS, __VA_ARGS__))
+
+} // namespace runtime
+} // namespace executorch
+
+namespace executorch {
+namespace aten {
+#ifdef USE_ATEN_LIB
+using ::at::elementSize;
+#else  // USE_ATEN_LIB
+using ::executorch::runtime::elementSize;
+#endif // USE_ATEN_LIB
+} // namespace aten
+} // namespace executorch
+
+namespace torch {
+namespace executor {
+// TODO(T197294990): Remove these deprecated aliases once all users have moved
+// to the new `::executorch` namespaces.
+using ::executorch::runtime::can_cast;
+using ::executorch::runtime::canCast;
+using ::executorch::runtime::convert;
+using ::executorch::runtime::CppTypeToScalarType;
+using ::executorch::runtime::elementSize;
+using ::executorch::runtime::is_barebones_unsigned_type;
+using ::executorch::runtime::is_bits_type;
+using ::executorch::runtime::is_complex_type;
+using ::executorch::runtime::is_float8_type;
+using ::executorch::runtime::is_integral_type;
+using ::executorch::runtime::is_qint_type;
+using ::executorch::runtime::isBitsType;
+using ::executorch::runtime::isComplexType;
+using ::executorch::runtime::isFloatingType;
+using ::executorch::runtime::isIntegralType;
+using ::executorch::runtime::isQIntType;
+using ::executorch::runtime::isRealHBType;
+using ::executorch::runtime::isRealHType;
+using ::executorch::runtime::isRealType;
+using ::executorch::runtime::isValid;
+using ::executorch::runtime::promote_types;
+using ::executorch::runtime::promoteTypes;
+using ::executorch::runtime::ScalarTypeToCppType;
+using ::executorch::runtime::toString;
+#if !defined(USE_ATEN_LIB)
+using ::executorch::runtime::is_floating_point;
+using ::executorch::runtime::is_reduced_floating_point;
+#endif
+namespace internal {
+using ::executorch::runtime::internal::B1;
+using ::executorch::runtime::internal::C2;
+using ::executorch::runtime::internal::C4;
+using ::executorch::runtime::internal::C8;
+using ::executorch::runtime::internal::F2;
+using ::executorch::runtime::internal::F4;
+using ::executorch::runtime::internal::F8;
+using ::executorch::runtime::internal::I1;
+using ::executorch::runtime::internal::I2;
+using ::executorch::runtime::internal::I4;
+using ::executorch::runtime::internal::I8;
+using ::executorch::runtime::internal::U1;
+} // namespace internal
+} // namespace executor
+} // namespace torch