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])