mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-12-24 13:28:13 +08:00
feat: select adapter id for DirectML
This commit is contained in:
@@ -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;
|
||||
|
||||
Reference in New Issue
Block a user