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

check dtypes first

Browse Source
This commit is contained in:
justheuristic 2022-09-17 23:13:23 +03:00
parent 140cdbe876
commit 9379df85d2
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 == torch.dtype assert out_bnb.dtype == out_torch.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()