diff options
author | Moritz Pflanzer <moritz.pflanzer@arm.com> | 2017-07-17 13:50:12 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-09-17 14:16:42 +0100 |
commit | d03b00acd71847fa2db1c5308c87d3b57c781bf9 (patch) | |
tree | 7c58283ca1a067428ccafd0f310c8c0e89800306 /tests/DatasetManager.h | |
parent | ee493ae23b8cd6de5a6c578cea34bccb478d2f64 (diff) | |
download | ComputeLibrary-d03b00acd71847fa2db1c5308c87d3b57c781bf9.tar.gz |
COMPMID-415: Fix dataset modes
Change-Id: I266e8a22890c914edb3335104f073e79d2bf0ad9
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/80766
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'tests/DatasetManager.h')
-rw-r--r-- | tests/DatasetManager.h | 101 |
1 files changed, 0 insertions, 101 deletions
diff --git a/tests/DatasetManager.h b/tests/DatasetManager.h deleted file mode 100644 index 2080203286..0000000000 --- a/tests/DatasetManager.h +++ /dev/null @@ -1,101 +0,0 @@ -/* - * Copyright (c) 2017 ARM Limited. - * - * SPDX-License-Identifier: MIT - * - * Permission is hereby granted, free of charge, to any person obtaining a copy - * of this software and associated documentation files (the "Software"), to - * deal in the Software without restriction, including without limitation the - * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or - * sell copies of the Software, and to permit persons to whom the Software is - * furnished to do so, subject to the following conditions: - * - * The above copyright notice and this permission notice shall be included in all - * copies or substantial portions of the Software. - * - * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR - * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, - * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE - * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER - * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, - * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE - * SOFTWARE. - */ -#ifndef ARM_COMPUTE_TEST_DATASETMANAGER -#define ARM_COMPUTE_TEST_DATASETMANAGER - -#include "arm_compute/core/TensorShape.h" -#include "framework/datasets/Datasets.h" - -#include <sstream> -#include <stdexcept> -#include <string> - -namespace arm_compute -{ -namespace test -{ -class DatasetManager final -{ -public: - enum class DatasetMode : unsigned int - { - ALL = 0, - PRECOMMIT = 1, - NIGHTLY = 2 - }; - - using ShapesDataset = framework::dataset::RangeDataset<std::vector<arm_compute::TensorShape>::const_iterator>; - - static DatasetManager &get(); - - void set_mode(DatasetMode mode); - - ShapesDataset shapesDataset() const; - -private: - DatasetManager() = default; - ~DatasetManager() = default; - - DatasetMode _mode{ DatasetMode::ALL }; -}; - -DatasetManager::DatasetMode dataset_mode_from_name(const std::string &name); - -inline ::std::stringstream &operator>>(::std::stringstream &stream, DatasetManager::DatasetMode &mode) -{ - std::string value; - stream >> value; - mode = dataset_mode_from_name(value); - return stream; -} - -inline ::std::stringstream &operator<<(::std::stringstream &stream, DatasetManager::DatasetMode mode) -{ - switch(mode) - { - case DatasetManager::DatasetMode::PRECOMMIT: - stream << "PRECOMMIT"; - break; - case DatasetManager::DatasetMode::NIGHTLY: - stream << "NIGHTLY"; - break; - case DatasetManager::DatasetMode::ALL: - stream << "ALL"; - break; - default: - throw std::invalid_argument("Unsupported dataset mode"); - } - - return stream; -} - -inline std::string to_string(const DatasetManager::DatasetMode &mode) -{ - std::stringstream stream; - stream << mode; - return stream.str(); -} -} // namespace test -} // namespace arm_compute -#endif /* ARM_COMPUTE_TEST_DATASETMANAGER */ |