Skip to content

Commit d107ac8

Browse files
openvino gpu device support
1 parent 3e63541 commit d107ac8

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

optimum_benchmark/backends/openvino/backend.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -79,6 +79,8 @@ def load_ovmodel_with_no_weights(self) -> None:
7979
def ovmodel_kwargs(self) -> Dict[str, Any]:
8080
kwargs = {}
8181

82+
kwargs["device"] = self.config.device
83+
8284
if self.config.export is not None:
8385
kwargs["export"] = self.config.export
8486

0 commit comments

Comments
 (0)