aboutsummaryrefslogtreecommitdiff
path: root/reference_model/src/generate/generate_utils.h
diff options
context:
space:
mode:
Diffstat (limited to 'reference_model/src/generate/generate_utils.h')
-rw-r--r--reference_model/src/generate/generate_utils.h10
1 files changed, 10 insertions, 0 deletions
diff --git a/reference_model/src/generate/generate_utils.h b/reference_model/src/generate/generate_utils.h
index e8e67bb..0239e98 100644
--- a/reference_model/src/generate/generate_utils.h
+++ b/reference_model/src/generate/generate_utils.h
@@ -55,6 +55,15 @@ struct DotProductInfo
std::array<int32_t, 2> kernel;
};
+/// \brief Pseudo random generator meta-data
+struct PseudoRandomInfo
+{
+ PseudoRandomInfo() = default;
+
+ int64_t rngSeed;
+ // TODO: Add range support
+};
+
/// \brief Generator configuration
struct GenerateConfig
{
@@ -65,6 +74,7 @@ struct GenerateConfig
int32_t inputPos;
tosa::Op opType;
DotProductInfo dotProductInfo;
+ PseudoRandomInfo pseudoRandomInfo;
};
/// \brief Parse the generator config when given in JSON form