checkStaticTypes Class — pytorch Architecture

Source Code

aten/src/ATen/core/op_registration/infer_schema.h lines 33–44

template <class... Types>
constexpr int checkStaticTypes() {
 // Give nice error messages for some of the common error cases.
 // Use a LOUD ERROR MESSAGE SO USERS SEE THE STATIC_ASSERT
 static_assert(std::conjunction_v<
     bool_t<!std::is_integral_v<Types> || std::is_same_v<Types, int8_t> || std::is_same_v<Types, int64_t> || std::is_same_v<Types, bool>>...
   >, "INVALID TYPE: Only int8_t, int64_t and bool are supported as an integral argument type");
 static_assert(std::conjunction_v<
     bool_t<!std::is_same_v<Types, float>>...
   >, "INVALID TYPE: float is not supported as an argument type, use double instead");
 return 0;
}