xref: /aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/TensorFactory.h (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1 //  Copyright © 2022 Apple Inc.
2 
3 #define AT_DISPATCH_MPS_TYPES(TYPE, NAME, ...)                          \
4   AT_DISPATCH_SWITCH(                                                   \
5       TYPE, NAME,                                                       \
6       AT_DISPATCH_CASE(at::ScalarType::Float, __VA_ARGS__)              \
7       AT_DISPATCH_CASE(at::ScalarType::Half, __VA_ARGS__)               \
8       AT_DISPATCH_CASE(at::ScalarType::Long, __VA_ARGS__)               \
9       AT_DISPATCH_CASE(at::ScalarType::Int, __VA_ARGS__)                \
10       AT_DISPATCH_CASE(at::ScalarType::Short, __VA_ARGS__)              \
11       AT_DISPATCH_CASE(at::ScalarType::Char, __VA_ARGS__)               \
12       AT_DISPATCH_CASE(at::ScalarType::Byte, __VA_ARGS__))
13