fz-mirrors/MaaAssistantArknights
0
0
Fork 0
mirror of https://github.com/MaaAssistantArknights/MaaAssistantArknights.git synced 2026-07-16 17:57:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: GPU acceleration (DX12/DirectML)

Browse Source 
Co-authored-by: SherkeyXD <253294679@qq.com>
This commit is contained in:
dantmnf
2024-05-30 01:56:21 +08:00
parent 3e5b6dd4fb
commit d6572b2c7e
26 changed files with 594 additions and 176 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/MaaCore/Assistant.cpp
View File

@@ -6,6 +6,7 @@
#include "Config/GeneralConfig.h"
#include "Config/Miscellaneous/OcrPack.h"
#include "Config/OnnxSessions.h"
#include "Config/ResourceLoader.h"
#include "Controller/Controller.h"
#include "Status.h"
@@ -40,12 +41,14 @@ bool ::AsstExtAPI::set_static_option(StaticOptionKey key, const std::string& val
case StaticOptionKey::CpuOCR: {
WordOcr::get_instance().use_cpu();
CharOcr::get_instance().use_cpu();
OnnxSessions::get_instance().use_cpu();
return true;
} break;
case StaticOptionKey::GpuOCR: {
int device_id = std::stoi(value);
WordOcr::get_instance().use_gpu(device_id);
CharOcr::get_instance().use_gpu(device_id);
OnnxSessions::get_instance().use_gpu(device_id);
return true;
} break;
default: