1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
|
#include <c10/core/impl/DeviceGuardImplInterface.h>
namespace c10 {
namespace impl {
// NOLINTNEXTLINE(modernize-avoid-c-arrays,cppcoreguidelines-avoid-c-arrays)
std::atomic<const DeviceGuardImplInterface*>
device_guard_impl_registry[static_cast<size_t>(
DeviceType::COMPILE_TIME_MAX_DEVICE_TYPES)];
DeviceGuardImplRegistrar::DeviceGuardImplRegistrar(
DeviceType type,
const DeviceGuardImplInterface* impl) {
device_guard_impl_registry[static_cast<size_t>(type)].store(impl);
}
} // namespace impl
} // namespace c10
|
