Home
last modified time | relevance | path

Searched refs:privateuse1_backend_name_set (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/c10/core/
DDeviceType.cpp122 static std::atomic<bool> privateuse1_backend_name_set; variable
130 privateuse1_backend_name_set.load(std::memory_order_acquire); in get_privateuse1_backend()
144 !privateuse1_backend_name_set.load() || in register_privateuse1_backend()
159 privateuse1_backend_name_set.store(true, std::memory_order_relaxed); in register_privateuse1_backend()
163 return privateuse1_backend_name_set.load(std::memory_order_acquire); in is_privateuse1_backend_registered()