diff options
| author | Jeremy Johnson <jeremy.johnson@arm.com> | 2023-11-15 15:52:06 +0000 |
|---|---|---|
| committer | Jeremy Johnson <jeremy.johnson@arm.com> | 2023-11-23 14:09:14 +0000 |
| commit | a015001dfbd0ed48caf54fd66b0509ee344a229e (patch) | |
| tree | 5f99a7d2d4aba2db2e672efb1168db961f99a544 /reference_model/src/generate/generate_pseudo_random.cc | |
| parent | 0bbd8bcfb20ec834f18d0bb89fc69ba4e92b3019 (diff) | |
| download | reference_model-a015001dfbd0ed48caf54fd66b0509ee344a229e.tar.gz | |
Main Compliance testing support for COMPARISON ops
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: Id6229cfaccad866b110630119eb045dbf6453bf5
Diffstat (limited to 'reference_model/src/generate/generate_pseudo_random.cc')
| -rw-r--r-- | reference_model/src/generate/generate_pseudo_random.cc | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/reference_model/src/generate/generate_pseudo_random.cc b/reference_model/src/generate/generate_pseudo_random.cc index 78013eb..d8d2288 100644 --- a/reference_model/src/generate/generate_pseudo_random.cc +++ b/reference_model/src/generate/generate_pseudo_random.cc @@ -107,9 +107,16 @@ bool generateFP32(const TosaReference::GenerateConfig& cfg, void* data, size_t s float* a = reinterpret_cast<float*>(data); const auto T = TosaReference::numElementsFromShape(cfg.shape); + const bool comparisonOp = + (cfg.opType == Op::Op_EQUAL) || (cfg.opType == Op::Op_GREATER_EQUAL) || (cfg.opType == Op::Op_GREATER); for (auto t = 0; t < T; ++t) { a[t] = generator->getRandomFloat(); + if (comparisonOp && (t % 4 == 0)) + { + // Set every 4th value to 0 to enable better comparison testing + a[t] = 0.f; + } } return true; } |