Fix: Use memory_order_relaxed instead of memory_order_relaxed (#159105)

Addresses #159074 by using `memory_order_release` instead of `memory_order_relaxed` here:

9c10760662/c10/core/DeviceType.cpp (L161)

Pull Request resolved: https://github.com/pytorch/pytorch/pull/159105
Approved by: https://github.com/colesbury
This commit is contained in:
Sameer
2025-07-25 19:38:59 +00:00
committed by PyTorch MergeBot
parent ba949c54a7
commit 51cd6697cd

View File

@ -158,7 +158,7 @@ void register_privateuse1_backend(const std::string& backend_name) {
privateuse1_backend_name = backend_name;
// Invariant: once this flag is set, privateuse1_backend_name is NEVER written
// to.
privateuse1_backend_name_set.store(true, std::memory_order_relaxed);
privateuse1_backend_name_set.store(true, std::memory_order_release);
}
bool is_privateuse1_backend_registered() {