From 9e64563a55f414b3d2d4529ceeffb895d19ce050 Mon Sep 17 00:00:00 2001 From: kprokofi Date: Sun, 17 Mar 2024 07:52:16 +0900 Subject: [PATCH] change device --- tests/perf/test_classification.py | 6 +++--- tests/perf/test_detection.py | 2 +- tests/perf/test_instance_segmentation.py | 4 ++-- 3 files changed, 6 insertions(+), 6 deletions(-) diff --git a/tests/perf/test_classification.py b/tests/perf/test_classification.py index 19abc226e36..873fd82c9ef 100644 --- 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 index 70e0cd2892b..51c6803100b 100644 --- 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 index d3afcd4310b..b71405484e9 100644 --- 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,