aboutsummaryrefslogtreecommitdiff
path: root/reference_model/src/generate/generate_dot_product.cc
diff options
context:
space:
mode:
Diffstat (limited to 'reference_model/src/generate/generate_dot_product.cc')
-rw-r--r--reference_model/src/generate/generate_dot_product.cc5
1 files changed, 5 insertions, 0 deletions
diff --git a/reference_model/src/generate/generate_dot_product.cc b/reference_model/src/generate/generate_dot_product.cc
index 1d2325f..cbfac4b 100644
--- a/reference_model/src/generate/generate_dot_product.cc
+++ b/reference_model/src/generate/generate_dot_product.cc
@@ -56,6 +56,11 @@ bool generateMatMul(const TosaReference::GenerateConfig& cfg,
void* data,
size_t size)
{
+ if (cfg.dataType != DType::DType_FP32)
+ {
+ WARNING("[Generator][DP][MatMul] Only supports FP32.");
+ return false;
+ }
if (cfg.shape.size() != 3)
{
WARNING("[Generator][DP][MatMul] Tensor shape expected 3 dimensions.");