mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
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:
@ -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() {
|
||||
|
Reference in New Issue
Block a user