apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-12-24 13:28:13 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

feat: select adapter id for DirectML

Browse Source
This commit is contained in:
MistEO
2024-11-20 14:37:26 +08:00
parent 3bb05ac574
commit 2507a172f8
3 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
fastdeploy/runtime/runtime_option.cc
View File

@@ -141,7 +141,10 @@ void RuntimeOption::UseAscend() {
paddle_lite_option.device = device;
}
void RuntimeOption::UseDirectML() { device = Device::DIRECTML; }
void RuntimeOption::UseDirectML(int adapter_id) {
device = Device::DIRECTML;
device_id = adapter_id;
}
void RuntimeOption::UseSophgo() {
device = Device::SOPHGOTPUD;