diff options
author | Jerry Ge <jerry.ge@arm.com> | 2023-04-11 00:05:02 +0000 |
---|---|---|
committer | Jerry Ge <jerry.ge@arm.com> | 2023-04-20 22:53:37 +0000 |
commit | a793f4645d2c04543572de4d0bc84bf0a3689604 (patch) | |
tree | cfa8ff162c9315f079682c0913110ad25eb22cad /reference_model/include/func_config.h | |
parent | 714aa6039a7e3585bf81ac90ce301767c08295af (diff) | |
download | reference_model-a793f4645d2c04543572de4d0bc84bf0a3689604.tar.gz |
Add level checking to TOSA Ref model
Signed-off-by: Jerry Ge <jerry.ge@arm.com>
Change-Id: I5689d7c6b902a319a68fa4628b59e0bcc23aeca4
Diffstat (limited to 'reference_model/include/func_config.h')
-rw-r--r-- | reference_model/include/func_config.h | 18 |
1 files changed, 17 insertions, 1 deletions
diff --git a/reference_model/include/func_config.h b/reference_model/include/func_config.h index d9b51d5..c1f8ef6 100644 --- a/reference_model/include/func_config.h +++ b/reference_model/include/func_config.h @@ -1,5 +1,5 @@ -// Copyright (c) 2020-2022, ARM Limited. +// Copyright (c) 2020-2023, ARM Limited. // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. @@ -19,6 +19,18 @@ #include <iostream> #include <stdio.h> +struct tosa_level_t { + int32_t MAX_RANK = 0; + int32_t MAX_KERNEL = 0; + int32_t MAX_STRIDE = 0; + int32_t MAX_SCALE = 0; + + bool operator!=(const tosa_level_t &rhs) { + return !(MAX_RANK == rhs.MAX_RANK && MAX_KERNEL == rhs.MAX_KERNEL && + MAX_STRIDE == rhs.MAX_STRIDE && MAX_SCALE == rhs.MAX_SCALE); + } +}; + struct func_config_t { std::string operator_fbs = "tosa.fbs"; @@ -37,6 +49,10 @@ struct func_config_t uint32_t dump_intermediates = 0; std::string fp_format = "0.5"; bool float_is_big_endian = false; // Set in arith_util.h by float_is_big_endian() + + tosa_level_t tosa_level; + static constexpr tosa_level_t EIGHTK = { 6, 8192, 8192, 64 }; + static constexpr tosa_level_t NONE = { 0, 0, 0, 0 }; }; #endif |