change device

openvinotoolkit · Mar 16, 2024 · 9e64563 · 9e64563
1 parent 1292d1c
commit 9e64563
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/tests/perf/test_classification.py b/tests/perf/test_classification.py
@@ -114,7 +114,7 @@ def test_perf(
         fxt_dataset: Benchmark.Dataset,
         fxt_benchmark: Benchmark,
     ):
-        fxt_benchmark.accelerator = "gpu"
+        fxt_benchmark.accelerator = "xpu"
         self._test_perf(
             model=fxt_model,
             dataset=fxt_dataset,
@@ -196,7 +196,7 @@ def test_perf(
         fxt_dataset: Benchmark.Dataset,
         fxt_benchmark: Benchmark,
     ):
-        fxt_benchmark.accelerator = "gpu"
+        fxt_benchmark.accelerator = "xpu"
         self._test_perf(
             model=fxt_model,
             dataset=fxt_dataset,
@@ -276,7 +276,7 @@ def test_perf(
         fxt_dataset: Benchmark.Dataset,
         fxt_benchmark: Benchmark,
     ):
-        fxt_benchmark.accelerator = "gpu"
+        fxt_benchmark.accelerator = "xpu"
         self._test_perf(
             model=fxt_model,
             dataset=fxt_dataset,

diff --git a/tests/perf/test_detection.py b/tests/perf/test_detection.py
@@ -118,7 +118,7 @@ def test_perf(
         fxt_dataset: Benchmark.Dataset,
         fxt_benchmark: Benchmark,
     ):
-        fxt_benchmark.accelerator = "gpu"
+        fxt_benchmark.accelerator = "xpu"
         self._test_perf(
             model=fxt_model,
             dataset=fxt_dataset,

diff --git a/tests/perf/test_instance_segmentation.py b/tests/perf/test_instance_segmentation.py
@@ -100,7 +100,7 @@ def test_perf(
         fxt_dataset: Benchmark.Dataset,
         fxt_benchmark: Benchmark,
     ):
-        fxt_benchmark.accelerator = "gpu"
+        fxt_benchmark.accelerator = "xpu"
         self._test_perf(
             model=fxt_model,
             dataset=fxt_dataset,
@@ -183,7 +183,7 @@ def test_perf(
         fxt_dataset: Benchmark.Dataset,
         fxt_benchmark: Benchmark,
     ):
-        fxt_benchmark.accelerator = "gpu"
+        fxt_benchmark.accelerator = "xpu"
         self._test_perf(
             model=fxt_model,
             dataset=fxt_dataset,