From: Andrey Miroshnikov Date: Thu, 19 Oct 2023 09:25:17 +0000 (+0000) Subject: Replace flatten func with builtin reduce() X-Git-Url: https://git.libre-soc.org/?a=commitdiff_plain;h=be148639eadbf26b009092b7ac28b9d4d588009f;p=openpower-isa.git Replace flatten func with builtin reduce() --- diff --git a/src/openpower/decoder/isa/test_caller_svp64_matrix.py b/src/openpower/decoder/isa/test_caller_svp64_matrix.py index 6fd64a2c..fab9c2e9 100644 --- a/src/openpower/decoder/isa/test_caller_svp64_matrix.py +++ b/src/openpower/decoder/isa/test_caller_svp64_matrix.py @@ -44,10 +44,6 @@ def matmult_inner(a,b): return result -# Flatten list of lists matrix down to single list -def flatten(l): - return [item for sublist in l for item in sublist] - class DecoderTestCase(FHDLTestCase): @@ -83,22 +79,16 @@ class DecoderTestCase(FHDLTestCase): X = X1 Y = Y1 - expected = matmult_outer(X, Y) - expected2 = matmult_inner(X, Y) - expected = flatten(expected) - expected2 = flatten(expected2) - print("expected-matrix (outer):") - print(expected) - print("expected-matrix (inner):") - print(expected2) xf = reduce(operator.add, X) yf = reduce(operator.add, Y) - expected = reduce(operator.add, expected) - print("flattened X,Y,expected") + expected = reduce(operator.add, matmult_outer(X, Y)) + expected2 = reduce(operator.add, matmult_inner(X, Y)) + print("flattened X,Y,expected (outer), expected (inner)") print("\t", xf) print("\t", yf) print("\t", expected) + print("\t", expected2) # and create a linear result2, same scheme #result1 = [0] * (ydim1*xdim2)