diff options
Diffstat (limited to 'tests/benchmark_new/CL/SYSTEM')
-rw-r--r-- | tests/benchmark_new/CL/SYSTEM/AlexNet.cpp | 2 | ||||
-rw-r--r-- | tests/benchmark_new/CL/SYSTEM/LeNet5.cpp | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/tests/benchmark_new/CL/SYSTEM/AlexNet.cpp b/tests/benchmark_new/CL/SYSTEM/AlexNet.cpp index a0673b94df..90c15c536a 100644 --- a/tests/benchmark_new/CL/SYSTEM/AlexNet.cpp +++ b/tests/benchmark_new/CL/SYSTEM/AlexNet.cpp @@ -57,7 +57,7 @@ using CLAlexNetFixture = AlexNetFixture<ICLTensor, TEST_SUITE(SYSTEM_TEST) TEST_SUITE(CL) -REGISTER_FIXTURE_DATA_TEST_CASE(AlexNet, CLAlexNetFixture, +REGISTER_FIXTURE_DATA_TEST_CASE(AlexNet, CLAlexNetFixture, framework::DatasetMode::ALL, framework::dataset::combine(framework::dataset::make("Data type", DataType::F32), framework::dataset::make("Batches", { 1, 4, 8 }))); diff --git a/tests/benchmark_new/CL/SYSTEM/LeNet5.cpp b/tests/benchmark_new/CL/SYSTEM/LeNet5.cpp index 0ee7c0177a..b25fb685bf 100644 --- a/tests/benchmark_new/CL/SYSTEM/LeNet5.cpp +++ b/tests/benchmark_new/CL/SYSTEM/LeNet5.cpp @@ -51,7 +51,7 @@ using CLLeNet5Fixture = LeNet5Fixture<CLTensor, TEST_SUITE(SYSTEM_TEST) TEST_SUITE(CL) -REGISTER_FIXTURE_DATA_TEST_CASE(LeNet5, CLLeNet5Fixture, +REGISTER_FIXTURE_DATA_TEST_CASE(LeNet5, CLLeNet5Fixture, framework::DatasetMode::ALL, framework::dataset::make("Batches", { 1, 4, 8 })); TEST_SUITE_END() |