Skip to content

GitLab

Unverified Commit 7d075d10 authored by q.yao's avatar q.yao Committed by GitHub
Browse files

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

parent 264e170c
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 5 deletions
mmcv/ops/csrc/common/mps/MPSLibrary.mm
View file @ 7d075d10
#include "MPSLibrary.h" #include "MPSLibrary.h"
#include <c10/util/CallOnce.h>
#include "MPSDevice.h" #include "MPSDevice.h"
static std::unique_ptr<MPSLibraryManager> mps_library_manager; static std::unique_ptr<MPSLibraryManager> mps_library_manager=nullptr;
static c10::once_flag mpsdev_init;
MPSLibraryManager* MPSLibraryManager::getInstance() { MPSLibraryManager* MPSLibraryManager::getInstance() {
c10::call_once(mpsdev_init, [] { if(!mps_library_manager)
mps_library_manager = std::unique_ptr<MPSLibraryManager>(new MPSLibraryManager()); mps_library_manager = std::unique_ptr<MPSLibraryManager>(new MPSLibraryManager());
});
return mps_library_manager.get(); return mps_library_manager.get();
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment