mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-28 02:42:21 +08:00
feat: select adapter id for DirectML
This commit is contained in:
@@ -98,7 +98,7 @@ bool OrtBackend::BuildOption(const OrtBackendOption& option) {
|
||||
"DML", ORT_API_VERSION, reinterpret_cast<const void**>(&ortDmlApi));
|
||||
OrtStatus* onnx_dml_status =
|
||||
ortDmlApi->SessionOptionsAppendExecutionProvider_DML(session_options_,
|
||||
0);
|
||||
option_.device_id);
|
||||
if (onnx_dml_status != nullptr) {
|
||||
FDERROR
|
||||
<< "DirectML is not support in your machine, the program will exit."
|
||||
|
||||
@@ -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;
|
||||
|
||||
@@ -82,7 +82,7 @@ struct FASTDEPLOY_DECL RuntimeOption {
|
||||
void UseAscend();
|
||||
|
||||
/// Use onnxruntime DirectML to inference
|
||||
void UseDirectML();
|
||||
void UseDirectML(int adapter_id = 0);
|
||||
|
||||
/// Use Sophgo to inference
|
||||
void UseSophgo();
|
||||
|
||||
Reference in New Issue
Block a user