Skip to content

Commit 4500962

Browse files
collucaViviane Potocnik
authored and
Viviane Potocnik
committed
gemm: Extend datagen assertions
1 parent fcfa334 commit 4500962

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

sw/blas/gemm/scripts/datagen.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,8 @@ def validate_config(self, gemm_fp, parallelize_m,
7979
' not supported for FP64 and FP32'
8080
assert not (dtype == 1 and impl == "opt"), 'FP8 not supported in' \
8181
' optimized implementation' \
82-
' (switch to opt_ex)'
82+
' (switch to OPT_EX)'
83+
assert dtype == 8 or beta == 0, 'beta != 0 supported only in FP64'
8384

8485
def emit_header(self, **kwargs):
8586
header = [super().emit_header()]

0 commit comments

Comments
 (0)