1#pragma once
2#include <c10/macros/Macros.h>
3#include <cstring>
4
5namespace c10 {
6namespace detail {
7
8template <typename T>
9struct LoadImpl {
10 C10_HOST_DEVICE static T apply(const void* src) {
11 return *reinterpret_cast<const T*>(src);
12 }
13};
14
15template <>
16struct LoadImpl<bool> {
17 C10_HOST_DEVICE static bool apply(const void* src) {
18 static_assert(sizeof(bool) == sizeof(char), "");
19 // NOTE: [Loading boolean values]
20 // Protect against invalid boolean values by loading as a byte
21 // first, then converting to bool (see gh-54789).
22 return *reinterpret_cast<const unsigned char*>(src);
23 }
24};
25
26} // namespace detail
27
28template <typename T>
29C10_HOST_DEVICE T load(const void* src) {
30 return c10::detail::LoadImpl<T>::apply(src);
31}
32
33template <typename scalar_t>
34C10_HOST_DEVICE scalar_t load(const scalar_t* src) {
35 return c10::detail::LoadImpl<scalar_t>::apply(src);
36}
37
38} // namespace c10
39