gemm: Extend datagen assertions

pulp-platform · May 8, 2024 · 3e171dd · 3e171dd
1 parent 1df7825
commit 3e171dd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/sw/blas/gemm/scripts/datagen.py b/sw/blas/gemm/scripts/datagen.py
@@ -79,7 +79,8 @@ def validate_config(self, gemm_fp, parallelize_m,
             ' not supported for FP64 and FP32'
         assert not (dtype == 1 and impl == "opt"), 'FP8 not supported in' \
             ' optimized implementation' \
-            ' (switch to opt_ex)'
+            ' (switch to OPT_EX)'
+        assert dtype == 8 or beta == 0, 'beta != 0 supported only in FP64'
 
     def emit_header(self, **kwargs):
         header = [super().emit_header()]