feat: select adapter id for DirectML

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

View File

@@ -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();