mirrors
/
mmcv
mirror of https://github.com/open-mmlab/mmcv.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[Fix] Fix MPS ops for torch1.12+ ()

pull/2371/head
q.yao 2022-09-29 15:02:37 +08:00 committed by Zaida Zhou
parent 1eef4f5a4e
commit f48975b2e3
1 changed files with 2 additions and 5 deletions
mmcv/ops/csrc/common/mps

7
mmcv/ops/csrc/common/mps/MPSLibrary.mm
View File

@ -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();
}