mrq/bitsandbytes-rocm
1
0
Fork 1
Code Issues 1 Pull Requests Packages Projects Releases Wiki Activity

Initial.

Browse Source
This commit is contained in:
Tim Dettmers 2023-05-02 07:53:29 -07:00
parent 9192c9de64
commit 9aa232cc39
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/test_functional.py
View File

@ -2406,6 +2406,7 @@ def test_cutlass3_gemm(dtype):
# #assert False, 'ERROR' # #assert False, 'ERROR'
c = int(C1.numel()*0.00125*(dim/256))+1 c = int(C1.numel()*0.00125*(dim/256))+1
assert_all_approx_close(C1, C2, 1e-5, 0.01, count=c) assert_all_approx_close(C1, C2, 1e-5, 0.01, count=c)
print('') print('')
print(dim, sum(errs)/len(errs)/math.sqrt(dim)) print(dim, sum(errs)/len(errs)/math.sqrt(dim))