deftdawg/bitsandbytes-rocm
1
0
Fork 0
forked from mrq/bitsandbytes-rocm
Code Pull Requests Activity

clearer assertions

Browse Source
This commit is contained in:
justheuristic 2022-09-17 23:22:04 +03:00
parent 9379df85d2
commit e29c5f5c41
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_autograd.py
View File

@ -367,7 +367,7 @@ def test_matmullt(
if has_bias: if has_bias:
out_torch += bias out_torch += bias
assert out_bnb.dtype == out_torch.dtype assert out_bnb.dtype == A.dtype, f"bnb matmullt received {A.dtype} but returned {out_bnb.dtype}"
n = out_bnb.numel() n = out_bnb.numel()
err = torch.abs(out_bnb - out_torch).mean().item() err = torch.abs(out_bnb - out_torch).mean().item()