#pragma once

#include <ATen/core/DeprecatedTypeProperties.h>

#include <ATen/core/Tensor.h>

#include <ATen/record_function.h>

#include <c10/macros/Macros.h>

#include <c10/util/Exception.h>

#include <c10/util/Half.h>

#include <c10/util/Metaprogramming.h>

#include <c10/util/complex.h>

#include <c10/util/string_view.h>

namespace minkowski {

#define MINK_PRIVATE_CASE_TYPE_USING_HINT(NAME, enum_type, type, HINT, ...) \

case enum_type: { \

using HINT = type; \

return __VA_ARGS__(); \

}

#define MINK_PRIVATE_CASE_TYPE(NAME, enum_type, type, HINT, ...) \

MINK_PRIVATE_CASE_TYPE_USING_HINT(NAME, enum_type, type, HINT, __VA_ARGS__)

#define MINK_DISPATCH_INTEGER_TYPES(TYPE, HINT, NAME, ...) \

[&] { \

const auto &the_type = TYPE; \

/* don't use TYPE again in case it is an expensive or side-effect op */ \

at::ScalarType _it = ::detail::scalar_type(the_type); \

switch (_it) { \

MINK_PRIVATE_CASE_TYPE(NAME, at::ScalarType::Int, int32_t, HINT, \

__VA_ARGS__) \

MINK_PRIVATE_CASE_TYPE(NAME, at::ScalarType::Long, int64_t, HINT, \

__VA_ARGS__) \

default: \

AT_ERROR(#NAME, " not implemented for '", toString(_it), "'"); \

} \

}()

} // namespace minkowski