From f48975b2e386a5d4c3e2d81b2a3a86f7bc1e0fc0 Mon Sep 17 00:00:00 2001
From: "q.yao" <streetyao@live.com>
Date: Thu, 29 Sep 2022 15:02:37 +0800
Subject: [PATCH] [Fix] Fix MPS ops for torch1.12+ (#2302)

---
 mmcv/ops/csrc/common/mps/MPSLibrary.mm | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

diff --git a/mmcv/ops/csrc/common/mps/MPSLibrary.mm b/mmcv/ops/csrc/common/mps/MPSLibrary.mm
index 1a3d635ca..99addc7e2 100644
--- a/mmcv/ops/csrc/common/mps/MPSLibrary.mm
+++ b/mmcv/ops/csrc/common/mps/MPSLibrary.mm
@@ -1,14 +1,11 @@
 #include "MPSLibrary.h"
-#include <c10/util/CallOnce.h>
 #include "MPSDevice.h"
 
-static std::unique_ptr<MPSLibraryManager> mps_library_manager;
-static c10::once_flag mpsdev_init;
+static std::unique_ptr<MPSLibraryManager> mps_library_manager=nullptr;
 
 MPSLibraryManager* MPSLibraryManager::getInstance() {
-  c10::call_once(mpsdev_init, [] {
+  if(!mps_library_manager)
     mps_library_manager = std::unique_ptr<MPSLibraryManager>(new MPSLibraryManager());
-  });
   return mps_library_manager.get();
 }