diff options
Diffstat (limited to 'reference_model/src/generate/generate_dot_product.cc')
-rw-r--r-- | reference_model/src/generate/generate_dot_product.cc | 5 |
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."); |