generate_matrix.py: Default to range, fp32

This commit is contained in:
Hansung Kim
2024-09-07 17:40:21 -07:00
parent a967c262b1
commit 863e92a85e

View File

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