wu-arch/kernels
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Write expected_C to a binary file

Browse Source
This commit is contained in:
Hansung Kim
2024-05-03 17:27:25 -07:00
parent 5bd25985c6
commit b4c812f9f8
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/kernel/tensor/check_correctness.py
View File

@@ -86,10 +86,15 @@ expected_A = expected['A_array']
expected_B = expected['B_array']
expected_C = expected['C_array']
expected_C = expected_C + expected_A @ expected_B
print('expected C:')
print(expected_C[0:8, 0:8])
print('got C:')
print(C_array[0:8, 0:8])
print('diff C:')
print((expected_C - C_array)[0:8, 0:8])
expected_C.astype('float32').tofile("c_expected.bin")
assert np.allclose(expected_A, A_array)
assert np.allclose(expected_B, B_array)
assert np.allclose(expected_C, C_array)
assert np.allclose(expected_C, C_array)