diff options
author | evacha01 <evan.chandler@arm.com> | 2024-02-07 11:21:55 +0000 |
---|---|---|
committer | evacha01 <evan.chandler@arm.com> | 2024-03-07 12:06:38 +0000 |
commit | 9c96eefbaca6c85be79529bce7ff04fd7dfe3a0d (patch) | |
tree | 55647ee0216800b621bd0b27277c6f895929ef3d /reference_model/src/generate/generate_full_range.cc | |
parent | 6e1e2bc06bff785e87577f24064bbc846300f8fd (diff) | |
download | reference_model-9c96eefbaca6c85be79529bce7ff04fd7dfe3a0d.tar.gz |
FULL data gen mode for FP16
Signed-off-by: evacha01 <evan.chandler@arm.com>
Change-Id: I81bb322132daf25328a40342edc62d8e1db9edd6
Diffstat (limited to 'reference_model/src/generate/generate_full_range.cc')
-rw-r--r-- | reference_model/src/generate/generate_full_range.cc | 59 |
1 files changed, 59 insertions, 0 deletions
diff --git a/reference_model/src/generate/generate_full_range.cc b/reference_model/src/generate/generate_full_range.cc new file mode 100644 index 0000000..d2a89da --- /dev/null +++ b/reference_model/src/generate/generate_full_range.cc @@ -0,0 +1,59 @@ +// Copyright (c) 2024, ARM Limited. +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. + +#include "generate_full_range.h" +#include "half.hpp" + +namespace +{ + +template <typename DataType> +bool generate(const TosaReference::GenerateConfig& cfg, DataType* data, size_t size) +{ + const TosaReference::FullRangeInfo& frinfo = cfg.fullRangeInfo; + DataType value = frinfo.startVal; + + const auto T = TosaReference::numElementsFromShape(cfg.shape); + for (auto t = 0; t < T; ++t) + { + data[t] = value; + value++; + } + return true; +} +} // namespace + +namespace TosaReference +{ +bool generateFullRange(const GenerateConfig& cfg, void* data, size_t size) +{ + // Check we support the operator + if (cfg.opType == Op::Op_UNKNOWN) + { + WARNING("[Generator][PR] Unknown operator."); + return false; + } + + switch (cfg.dataType) + { + case DType::DType_FP16: { + uint16_t* outData = reinterpret_cast<uint16_t*>(data); + return generate(cfg, outData, size); + } + default: + WARNING("[Generator][PR] Unsupported type."); + return false; + } +} +} // namespace TosaReference
\ No newline at end of file |