Skip to content

GitLab

Commit 9aa232cc authored by Tim Dettmers's avatar Tim Dettmers
Browse files

Initial.

parent 9192c9de
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tests/test_functional.py
View file @ 9aa232cc
...@@ -2406,6 +2406,7 @@ def test_cutlass3_gemm(dtype): ...@@ -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))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment