From 863e92a85e61ff74820f6ce1926c061ea4c76c76 Mon Sep 17 00:00:00 2001 From: Hansung Kim Date: Sat, 7 Sep 2024 17:40:21 -0700 Subject: [PATCH] generate_matrix.py: Default to range, fp32 --- tests/kernel/tensor/generate_matrix.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/kernel/tensor/generate_matrix.py b/tests/kernel/tensor/generate_matrix.py index 796a6ea9..d54ece46 100644 --- a/tests/kernel/tensor/generate_matrix.py +++ b/tests/kernel/tensor/generate_matrix.py @@ -46,7 +46,7 @@ def pack_fp16_by_row(array): if __name__ == "__main__": M, N, K = parse_mnk() - rand = True + rand = False if not rand: A_array = np.arange(M * K).reshape([M, K]) B_array = np.arange(K * N).reshape([K, N]) @@ -77,7 +77,7 @@ if __name__ == "__main__": np.savez("abc", A_array=A_array, B_array=B_array, C_array=C_array) - fp16 = True + fp16 = False if fp16: A_packed = pack_fp16_by_row(A_array) AT_packed = A_packed.transpose([1, 0, 2])